No test.py?

Question

No test.py?

Xuefei98 opened this issue 2 years ago · comments

I was about to testing the pretrained model on the dataset that I am planning on using but didn't see a test..py available

sithu3 · Answer 1 · Thu Jan 27 2022 10:16:03 GMT+0800 (China Standard Time)

You can use tools/infer.py

Xuefei98 · Answer 2 · Thu Jan 27 2022 10:19:15 GMT+0800 (China Standard Time)

My bad! Sorry i thought the name would be test.py. Thanks! But another problem I have is when I run the infer.py with configuration file ade20k.yml and test mode segformer.b3.ade.pth. It returns size mismatch error. But segformer.b3.ade.pth is the default model that the file has.

Xuefei98 · Answer 3 · Thu Jan 27 2022 10:20:14 GMT+0800 (China Standard Time)

size mismatch for decode_head.linear_pred.weight: copying a param with shape torch.Size([150, 768, 1, 1]) from checkpoint, the shape in current model is torch.Size([150, 256, 1, 1]).

Xuefei98 · Answer 4 · Thu Jan 27 2022 10:22:46 GMT+0800 (China Standard Time)

SOLUTION: VARIANT need to be changed too.