Tensorflow + MobileNetv2_SSD 训练 使用TransferLearning实现环视图像的角点检测——Tensorflow+MobileNetv2_SSDtensorflow+ssd_mobilenet实现目标检测的训练

参考链接1

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

参考链接2:

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