-
Notifications
You must be signed in to change notification settings - Fork 150
question about test on one class choice #45
Comments
If your model is trained with the full dataset, you can't test it with a single class in the test set. If you really want to test its performance in one class, see #10. |
Now I want to check the Miou, ACC and other information of each class. Is there any convenient way? |
I think it may be impossible to use existing functions to calculate the mAcc and mIoU of a single class if your model is trained with full dataset. The label prediction of each point can be in anywhere of 50 labels. You can write some code to calculate them manually. You can still calculate Acc by using |
Then I can calculate the Miou of each class separately in the prediction, which is equivalent to modifying the calculation_ shape_ IOU, right? |
Yes |
Thank you very much |
I directly train the full dataset, then use this trained model to test on one class choice
but there was a problem:
RuntimeError: Error(s) in loading state_dict for DataParallel:
size mismatch for module.conv11.weight: copying a param with shape torch.Size([50, 256, 1]) from checkpoint, the shape in current model is torch.Size([4, 128, 1]).
I hope you can answer. Thank you very much!
The text was updated successfully, but these errors were encountered: