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