Uses Chainer, NumPy, OpenCV.
Installation:
$ sudo pip install chainer opencv-python numpy
Uses PyTorch, Ignite, Numpy, OpenCV.
Installation:
PyTorch: see the official document.
$ sudo pip install pytorch-ignite opencv-python numpy
$ python train-chainer.py
$ python train-pytorch.py
You can specify number of epochs and GPU (if available).
$ python train-xx.py -e epochs -g GPUID
Outputs of the trained model w.r.t test data are saved after training.