一、导出模型
在E:\models\research\object_detection文件夹下面运行
python export_inference_graph.py \ --input_type image_tensor \ --pipeline_config_path training/ssd_mobilenet_v1_coco.config \ --trained_checkpoint_prefix training/model.ckpt-XXXXX \ --output_directory cup_detection
–trained_checkpoint_prefix training/model.ckpt-XXXXX 代码中XXXXX可以在training文件夹下面找一个最大的数输入进去
–output_directory cup_detection 这是模型保存路径,训练好的模型放在object_detection文件夹下cup_detection中
二、验证
在E:\models\research\object_detection文件夹下面运行jupyter notebook 打开object_detection_tutorial.ipynb
修改路径和识别数目
要验证的图片要统一重命名为imagex.jpg 可以识别多个图片
参考博客:
https://blog.csdn.net/chenmaolin88/article/details/79357263