search

Reagan1311 / DABNet

Depth-wise Asymmetric Bottleneck for Real-time Semantic Segmentation (BMVC2019)
https://github.com/Reagan1311/DABNet
MIT License
140 stars 36 forks source link

How to modify predict.py to accept one image? #17

Closed kaishijeng closed 4 years ago

kaishijeng commented 5 years ago

I like to use predict.py to do prediction of one test image. Do you know how to do it? I am not familiar with pytorch dataset.

Thanks,

Reagan1311 commented 4 years ago

If you want to evaluate the result on one image, you should use open-cv or PIL to read the image and send it to the model, so you need to change the dataloader from cityscapes dataset to one specific image.