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

(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

原文地址:https://www.cnblogs.com/vincent212212/p/13293248.html