fast-rcnn win10 tensorflow部署
1、下載代碼https://github.com/chde222/Faster-RCNN-TensorFlow-Python3
2、安裝所依賴包 pip install -r requirements.txt
或者單獨利用pip install cython
? ? ? ? ? ? ? ? ? ? ? pip install easydict
3、在 ./data/coco/pythonAPI 下打開cmd運行:
????? python setup.py build_ext --inplace
????? python setup.py build_ext install
?????? 在./lib/utils下打開cmd 運行:
????????????????????? python setup.py build_ext install
?4、下載voc2007數據集。
數據集使用的是VOC2007,下載地址:
http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCdevkit_08-Jun-2007.tar
由于被墻,可以下載百度云盤的數據集,鏈接:https://pan.baidu.com/s/1Y_RzqLvW4CAzTEq4ICFVUA?,提取碼:m9dl
在data下新建VOCDevkit2007文件夾。然后
下載文件后,將 VOCtrainval_06-Nov-2007.tar 解壓后的文件夾VOC2007放在data/VOCDevkit2007/ 路徑下。
5、下載vgg16.ckpt 數據集。
VGG16模型的下載地址:http://download.tensorflow.org/models/vgg_16_2016_08_28.tar.gz,也可去百度云盤下載,
鏈接:https://pan.baidu.com/s/11Ty10NJ-rgXkkvM92SVVKw ,提取碼:d2jz
下載后將文件重命名為:vgg16.ckpt? 然后在data下建立imagenet_weights文件夾,將vgg16.ckpt放入
6、打開pycharm。 運行train.py 文件即可
參考自https://blog.csdn.net/ytusdc/article/details/80255920
https://blog.csdn.net/kellyroslyn/article/details/92159004
7、如果跑demo.py的話,顯示下面的錯誤。
??? 這個地方入坑很久,下載的vgg16模型里就是沒有這個文件的,查了好久才知道,這應該是新版TensorFlow模型訓練時候產生的文件。作者沒有提供,沒有辦法,只有自己訓練。通過上一步的訓練能夠得到模型。把這些模型都放在這個output\vgg16\voc_2007_trainval+voc_2012_trainval\default路徑下。out目錄跟demo.py同目錄。
?? demo.py 默認使用的是res101,而我們現在用的是vgg16模型,所以更改demo.py 第108行,deault = vgg16。還有就是模型文件的一些命名可能跟demo.py 文件中的名字不一致,導致找不到文件,自己更改模型文件名,或者demo.py即可。
https://blog.csdn.net/u010554381/article/details/86233339
?
總結
以上是生活随笔為你收集整理的fast-rcnn win10 tensorflow部署的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: Python部分知识点
- 下一篇: sdr 软件_无线电爱好者系列-1.通过