The [`bts_test.py`](https://github.com/cogaplex-bts/bts/blob/master/pytorch/bts_test.py) in pytorch version of BTS only allows testing/inference on the KITTI or NYU V2 dataset. How do I perform inference on a single image?