im2rec.py をローカルで使えるようにする方法

SageMakerで画像認識をさせるのに、.lst が必要になり im2rec.py というスクリプトで変換をかけると便利なのですが、 jupyter notebook でデバッグ効率があまり良くないのでローカルで動かせるようにしました。

Step1. ソースをダウンロードしてこよう

パスの通ってるとこにソースをダウンロードしてください。
curl -O https://raw.githubusercontent.com/apache/incubator-mxnet/master/tools/im2rec.py

Step2. ライブラリをインストールしよう

im2rec.py が、　mxnet と python-opencv を利用しているので、その二つをpipでいれます。

dependency.sh

pip install mxnet
pip install opencv-python

自分の場合

python3 を利用しているのでこんな感じです。

dependency.sh

pip3 install mxnet
pip3 install opencv-python
python3 im2rec.py コマンド色々

Tips

lstファイルを作る

prefix_of_lstとimage_root_pathはご自身の環境に合わせて変更してください。

python3 im2rec.py --list --recursive prefix_of_lst image_root_path

RecordIOフォーマットのファイルを作る

python3 im2rec.py --resize 256 --num-thread 16 lst_file_path image_root_path