經歷了無數坑,終於在電腦上成功運行了ssd算法。
主要參考的博客:http://m.blog.csdn.net/yexiaogu1104/article/details/77415990
http://www.cnblogs.com/zlslch/p/6984403.html
下面記錄下我的流程。
1.從 https://github.com/balancap/SSD-Tensorflow 上下載源碼並解壓;
2.下載模型 https://drive.google.com/uc?id=0B0qPCUZ-3YwWZlJaRTRRQWRFYXM&export=download
存放在SSD-Tensorflow-master/checkpoints/這個目錄下
3.在終端進入 SSD-Tensorflow-master下的notebook目錄,運行jupyter-notebook進入http://localhost:8888/tree服務器,打開 ssd_notebook.ipynb文件
4.改變path = '../demo/'
成自己的圖片所在路徑
5. 從頭到尾運行所有cell。
最終的運行結果:
下面記錄下我經歷的坑:
1.運行時報錯:
ValueError: Only know how to handle extensions: ['png']; with Pillow installed matplotlib can handle more images
錯誤原因:未安裝Pillow庫導致不能加載更多格式的圖片
解決:sudo pacman -S python-pillow
安裝Pillow庫
2.
這一步不必進行,我改了之后程序會出錯,提示沒有visualization模塊,不改反而能成功運行。
3.一定要安裝好所需要的環境:notebook pil 等等!!!