我想要按照示例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.lst
和my_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.idx
和my_data_val.rec
。
問題:我需要做什麼不同才能創建一個包含我自己的.png圖像的合適.rec
文件?
額外的細節:
我碼頭工人容器(mxnet /蟒:GPU)內部組件由上搬運工轂DMLC提供;他們還在他們的github頁面上提供了這個例子。數據通過容器中的共享目錄可用。所以這可能是一個碼頭問題。是什麼讓我有點擔心這是一個碼頭問題,我不得不爲了im2rec
能夠導入cv2 ...我本來希望提供容器的人會照顧這一點。