參考鏈接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