2017-04-20 13 views
1

hereに従い、MXnetを使用したトレーニングのために私自身のデータセットを作成しようとしています。例に指定されている私のデータが編成されています。MXnetでim2recを使用してpng画像のデータセットを作成する

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

チュートリアルでは、最初のステップは.recファイルを作成するには、再び(異なるオプションを)im2rec.pyを実行し、.LSTファイルを作成するために、im2rec.pyを実行することであると言います。 (予想通り)

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

これを実行した後、2つのファイルが作成され、my_data_train.lstmy_data_val.lst:I型.LSTファイルを作成します。 2つのファイルの合計行数は、yes/no/ディレクトリのファイル数を組み合わせた数と同じです。

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

これは(黙って)クラッシュし、その後数秒間実行し、:その後、私はim2recに使用して.recファイルを作成する2番目の時間を実行しよう。このプロセスでは、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をインポートできるようにするためには、pip install opencv-pythonにしなければならないということです...私はコンテナを提供している人々がこれを世話したと思ったでしょう。

答えて

関連する問題