tensorflow訓練自己的數據集

1、生成訓練集和測試集的csv文件 2、根據csv文件分別生成訓練集和測試集的tfrecord文件 3、在label_map.pbtxt中配置label標籤類別 4、在training文件夾下準備預訓練模型文件,可在導出的訓練模型models下找到自己的模型。 5、在object_detection》samples>configs>找到coco數據集的配置文件ssd_mobilenet_v1_co
相關文章
相關標籤/搜索