Hi, thank you for your work. I meet a problem when I run the evaluate.py. I change the input image size into (473,473) and raise the error:
500 images are loaded!
473 4730 processd
Need 6 x 3 prediction tiles @ stride 316 px
Predicting tile 1
evaluate.py:119: UserWarning: volatile was removed and now has no effect. Use with torch.no_grad(): instead.
padded_prediction = net(Variable(torch.from_numpy(padded_img), volatile=True).cuda())
Predicting tile 2
Predicting tile 3
Predicting tile 4
Predicting tile 5
Predicting tile 6
Predicting tile 7
Predicting tile 8
Predicting tile 9
Predicting tile 10
Predicting tile 11
Predicting tile 12
Predicting tile 13
Predicting tile 14
Predicting tile 15
Predicting tile 16
Predicting tile 17
Predicting tile 18
Traceback (most recent call last):
File "evaluate.py", line 255, in
main()
File "evaluate.py", line 238, in main
seg_pred = seg_pred[ignore_index]
IndexError: boolean index did not match indexed array along dimension 0; dimension is 1024 but corresponding boolean dimension is 473
ZOUWENMIN
commented
3 years ago
Hi, thank you for your work. I meet a problem when I run the evaluate.py. I change the input image size into (473,473) and raise the error:
main()
File "evaluate.py", line 238, in main
seg_pred = seg_pred[ignore_index]
IndexError: boolean index did not match indexed array along dimension 0; dimension is 1024 but corresponding boolean dimension is 473
500 images are loaded! 473 4730 processd Need 6 x 3 prediction tiles @ stride 316 px Predicting tile 1 evaluate.py:119: UserWarning: volatile was removed and now has no effect. Use
with torch.no_grad():instead. padded_prediction = net(Variable(torch.from_numpy(padded_img), volatile=True).cuda()) Predicting tile 2 Predicting tile 3 Predicting tile 4 Predicting tile 5 Predicting tile 6 Predicting tile 7 Predicting tile 8 Predicting tile 9 Predicting tile 10 Predicting tile 11 Predicting tile 12 Predicting tile 13 Predicting tile 14 Predicting tile 15 Predicting tile 16 Predicting tile 17 Predicting tile 18 Traceback (most recent call last): File "evaluate.py", line 255, inHow can I solve this problem ???