由于TensorFlow2还没有出新的检测框架。所以这里采用的是TensorFlow1.15(为什么不用1.13以下的?因为1.13一下的tf不支持cuda10)
这是目标检测的地址
https://github.com/tensorflow/models/tree/master/research/object_detection
如何快速训练自己的数据集?这篇博客写的很清楚:
https://www.cnblogs.com/zongfa/p/9663649.html
tensorflow_object_detection训练:/anaconda3/envs/tensorflow15/bin/python model_main.py --pipeline_config_path=training/ssd_mobilenet_v1_coco.config --model_dir=training --num_train_steps=2000 --num_eval_steps=200 --alsologtostderr
转pb:/anaconda3/envs/tensorflow15/bin/python export_inference_
graph.py --input_type image_tensor --pipeline_config_path training/ssd_mobilenet_v1_coco.config --trained_checkpoint_prefix training/model.ckpt-2000 --output_directory zhangjn_detction
测试:运行,object_detection_tutorial.py
https://www.cnblogs.com/zongfa/p/9663649.html
参考博文:https://blog.youkuaiyun.com/monk1992/article/details/99873776
官方文档:https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/running_pets.md
当然,如果要采用C++调用这个pb模型。需要使用C++编译TensorFlow的源码。如何编译TensorFlow源码?这个问题网上有教程。本人也是搞了好久。这里可以编译2.0以上的版本。等我做完以后再更新详细。