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

predict.py issue #23

Closed rurubaobao closed 4 years ago

rurubaobao commented 4 years ago

hello,when I predict the camvid dataset,I have this issue: Traceback (most recent call last): File "C:/Users/lenovo/PycharmProjects/fr/gleason/DABNet-master/predict.py", line 112, in test_model(args) File "C:/Users/lenovo/PycharmProjects/fr/gleason/DABNet-master/predict.py", line 83, in test_model predict(args, testLoader, model) File "C:/Users/lenovo/PycharmProjects/fr/gleason/DABNet-master/predict.py", line 25, in predict for i, (input, size, name) in enumerate(test_loader): ValueError: too many values to unpack (expected 3)

Reagan1311 commented 4 years ago

If you want to predict the Camvid dataset, use the test.py, predict.py is used to output the prediction for official evaluation like Cityscapes dataset