Tensorflow + MobileNetv2_SSD 训练


参考链接1

使用TransferLearning实现环视图像的角点检测——Tensorflow+MobileNetv2_SSD

https://www.cnblogs.com/hayley111/p/12918678.html

参考链接2:

tensorflow+ssd_mobilenet实现目标检测的训练

https://www.cnblogs.com/gezhuangzhuang/p/10613468.html

 参考链接3:

https://blog.csdn.net/darkrabbit/article/details/81295025

 参考链接4:

https://blog.csdn.net/weixin_39750664/article/details/82502302

 

(base) dftc@dftc-ThinkStation-P720:~/yk/ssd/models/models-master/research$ python object_detection/builders/model_builder_test.py
Traceback (most recent call last):
  File "object_detection/builders/model_builder_test.py", line 18, in <module>
    from absl.testing import parameterized
ModuleNotFoundError: No module named 'absl'
解决方法,安装库的时候,安装conda install absl-py

 

训练速度慢,可能是cuda没有成功开启

pip install keras==2.2.4

删除cuda指引的软连接,重新连接到cuda10

在train.py加入几行代码(稍后补充)

再次运行即可

我batch_size=16,运行每步耗时0.2-0.3秒

batch_size=48,运行每步耗时0.6秒左右

 

 

 

 慢慢补充ing

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM