Tested with CUDA 10.0, Ubuntu 18.04, Python 3.6 with Conda and PyTorch 1.3. python inference.py --gpu: GPU index, if you have not GPU, just ignore it --output: output root (required) --data: data root ...