Chainer, OpenCV, NumPyを使います。
インストール:
$ sudo pip install chainer opencv-python numpy
PyTorch, Ignite, OpenCV, NumPyを使います。
インストール:
PyTorch: 公式を参照してください。
$ sudo pip install pytorch-ignite opencv-python numpy
$ python train-chainer.py
$ python train-pytorch.py
エポック数とGPU(使える場合)を指定できます。
$ python train-xx.py -e エポック数 -g GPUID
学習後、テストデータをモデルに入力した結果が画像として保存されます。