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