TensorFlow object_detection总结

由于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.csdn.net/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以上的版本。等我做完以后再更新详细。

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值