Can't run demo.py

[AmrutaAnalytics]

Trying to run pure inference part, Downgraded pytorch version to 1.5 (torch==1.5.0+cu101 torchvision==0.6.0+cu101), then Set CUDA_VISIBLE_DEVICES=0 python demo.py $EXPNAME --checkepoch 180 --img_root "path to image " , but getting following error in dataset/dataload.py

[princewang1994]

@AmrutaAnalytics Did you use your own dataset? Seems that your input image are gray-scale ones, If so, make it RGB format first.