例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.lst
とmy_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
にしなければならないということです...私はコンテナを提供している人々がこれを世話したと思ったでしょう。