search

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

different test batch_size #8

Closed lu123456789 closed 5 years ago

lu123456789 commented 5 years ago

Hello,i changed batch_size=1 to batch_size=batch_size when test. When i test the same model with different batch_sizes, why miou are different?

Reagan1311 commented 5 years ago

I think the main reason is the mIoU calculation function, the code is for batch size 1. Besides, you can't randomly set the batch_size, as the total number of images is 500 You should set the number that is divisible by 500, otherwise the last batch will contain the different number of image.

lu123456789 commented 5 years ago

thank you