2017-04-20 46 views
1

我想要按照示例here創建我自己的數據集,以便使用MXnet進行培訓。如示例指定我的數據組織:在MXnet中使用im2rec與png圖像創建數據集

/data 
    yes/ 
     file1.png 
     file2.png 
     ... 
    no/ 
     file1.png 
     file2.png 
     ... 

的教程說的第一步是運行im2rec.py創建.lst文件,然後再次運行im2rec.py(不同的選項)創建.REC文件。要創建.lst文件I型:

> python tools/im2rec.py my_data /data --list True --recursive True --train-ratio .75 --exts .png 

這樣做了以後,創建兩個文件(如預期),my_data_train.lstmy_data_val.lst。這兩個文件中的總行數與我組合的yes/no/目錄中的文件數相同。然後,我嘗試使用運行im2rec第二次創建.rec文件:

> python tools/im2rec.py my_data /data --resize 227 --num-thread 16 

它運行了幾秒鐘,然後(默默)崩潰。在此過程中,它會創建4個空文件:my_data_train.idx,my_data_train.rec,my_data_val.idxmy_data_val.rec

問題:我需要做什麼不同才能創建一個包含我自己的.png圖像的合適.rec文件?

額外的細節:

我碼頭工人容器(mxnet /蟒:GPU)內部組件由上搬運工轂DMLC提供;他們還在他們的github頁面上提供了這個例子。數據通過容器中的共享目錄可用。所以這可能是一個碼頭問題。是什麼讓我有點擔心這是一個碼頭問題,我不得不爲了im2rec能夠導入cv2 ...我本來希望提供容器的人會照顧這一點。

回答