gluon-cv
gluon-cv copied to clipboard
Add dim check while calculating mIoU&pixAcc for segmentation test
-
The test.py example mentioned at segmentation.rst does not match the arguments mentioned in test.py should probably be change to
python test.py --dataset ade20k --pretrained --model fcn --backbone resnet50 --eval
-
The dimension doesn't seem right when calculating the pixACC and mIoU The output is 3D and the target is 2D The safest way I could think of is to add a new axis if batch dim is missing. Might have a better way to deal with.
Result after modification (without training)
Apologies for the new spaces, it was automatically modified by Pycharm IDE
Job PR-991-1 is done.
Docs are uploaded to http://gluon-vision-staging.s3-website-us-west-2.amazonaws.com/PR-991/1/index.html
Code coverage of this PR: vs. Master:
python test.py --dataset ade20k --pretrained --model fcn --backbone resnet50 --eval
python test.py --dataset pascal_voc --pretrained --model fcn --backbone resnet101 --eval
Job PR-991-2 is done.
Docs are uploaded to http://gluon-vision-staging.s3-website-us-west-2.amazonaws.com/PR-991/2/index.html
Code coverage of this PR: vs. Master: