1.安装python环境
(2)Protobuf 安装与配置
protoc object_detection/proto/*.proto --python_out=.
models/research/ 及 models/research/slim 两个文件夹的完整目录添加,分号隔开。
2.
(1)打标签,通过程序将xml转csv
https://download.csdn.net/download/shanzhonglujie/10567280
(2)通过csv生成record
python generate_tfrecord.py --csv_input=data/tv_vehicle_labels.csv --output_path=data/train.record(训练)
(3)添加修改pbtxt文件,如data/tv_vehicle_detection.pbtxt
(4)修改配置文件,如train/ssd_mobilenet_v1_coco.config
(5)训练
python train.py --logtostderr --train_dir=object_detection/training/ --pipeline_config_path=c:/ssd_mobilenet_v1_coco.config
tensorboard --logdir=E:/models/research/object_detection/training/
(6)转化训练结果,生成pd文件
python export_inference_graph.py --input_type image_tensor --pipeline_config_path c:/ssd_mobilenet_v1_coco.config --trained_checkpoint_prefix E:/models/research/object_detection/training/model.ckpt-8309 --output_directory E:/models/research/object_detection/training/frozen_inference_graph.pb
(7)测试结果
3.问题
(1)ModuleNotFoundError: No module named 'object_detection'
解决方法:在research目录下运行命令行
(2)ImportError: No module named nets 或找不到net
解决方法:将reserch/slim目录下的BUILD文件删除,运行slim下的python setup.py install,如果再不行就运行一下research下的setup.py
(3)
解决方法:生成的record有问题,检查路径或重新生成
(4)缺少Pycocotools
解决方法:安装https://download.csdn.net/download/shanzhonglujie/10567294