1. 程式人生 > >Tensorflow框架下SSD演算法的樣例測試遇到的問題及成功執行!

Tensorflow框架下SSD演算法的樣例測試遇到的問題及成功執行!

主要參考部落格:https://blog.csdn.net/yexiaogu1104/article/details/77415990

步驟同上述部落格:

1. 下載模型ssd_300_vgg,存放在SSD-Tensorflow-master/checkpoints/這個目錄下 
2. 開啟終端,在主目錄 SSD-Tensorflow-master 下,執行cd notebooks/進入noteboo目錄,執行 jupyter-notebook進入http://localhost:8892/tree伺服器,開啟 ssd_notebook.ipynb.ipynb檔案 
3. 配置一些路徑:第4個cell中from notebooks import visualization

改成import notebooks,不改的話目錄結構不對,會報錯 
4. 使用自己的圖片,改變path = '../demo/'成自己的圖片所在路徑 

5. 從頭到尾執行所有cell即可。

遇到的問題及解決辦法:

1.cell6 一直是In[*]的狀態,這就會導致最後一個cell出現

NameError: name 'ssd_anchors' is not defined 

的錯誤;

2.路徑問題:ckpt_filename = '../checkpoints/VGG_VOC0712_SSD_300x300_iter_120000.ckpt',這樣就OK,但是有關鍵問題:一定要把下載的ssd_300_vgg放置到步驟二中的目錄下,注意不是資料夾,是裡面的兩個檔案!!!!!!!!!

3.使用自己的樣本:可以將要測試的圖片拷到demo資料夾下,注意更改最後一個cell中的程式碼就行,如圖:


-1 代表的是最後一張圖,可以更改數字測試任意張圖!

自己下的圖片測試結果直接上圖: