使用tensorflow訓練SSD(一):相關環境的配置


在使用TensorFlow進行目標檢測時,首先需要下載tensorflow object detection API模型,該模型的下載地址為https://github.com/tensorflow/models。 其次需要完成准備工作包括: 1)獲取protobuf 由於Object Detection API是使用protobufs來配置模型和訓練參數,這些文件以".proto"的擴展名放在models\research\object_detection\protos下。在使用框架之前,必須使用protobuf庫將其編譯成py文件才能正常使用。protobuf的下載地址為https://github.com/google/protobuf/releases中選擇對應的版本進行安裝。 2)編譯proto配置文件 在命令行中進入到models/research目錄下,執行如下命令:protoc.exe object_detection/protos/*.proto -- python_out=. 如果不顯示任何信息,則表明運行成功了,此時來到models/research/object_detection\protos文件下會發現生成了很多py文件。 3)檢測API是否正常 在完成上述操作以后,需要將models\research\slim中的nets文件夾復制到models\research下;models\research\object_detection\builders下的model_builder_test.py文件復制到models\research下。 然后運行research文件夾下的model_build_test.py文件,如果出現了OK字樣就表明API一切正常,可以正常使用了。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM