dbash / zerowaste

[CVPR 2022] ZeroWaste Dataset: Towards Deformable Object Segmentation in Cluttered Scenes
MIT License
42 stars 9 forks source link

Errors in Reproducing MIOU values #5

Closed aliman80 closed 1 year ago

aliman80 commented 1 year ago

Hi, Thank you for your wonderful work. This is the kind of work i was looking .I am trying to reproduce these results using deeplab but consistently my results are less in mIOU while for pACC they are ok. Similarly when i train model using RCNN my AP values are consistently less like i am g etting 17 instead of 22 and so on. Can you guide me if there is any hyperparamter which i needs to set or am i missing some thing. Thanking you in advance.

dbash commented 1 year ago

Hi Muhammad, thank you for your interest in our project.

First, have you tried using a pretrained checkpoint we provide for DeepLab with the corresponding configs, does this result in the same accuracy as reported in the paper? If not, there might be an issue with the config / loading the checkpoint.

If you are training from scratch, it is important to keep the same hyperparameters, especially the batch size.

aliman80 commented 1 year ago

Thank you for your response. Yes i treid your pretrained checkpoint it shows the same result as reported in the paper. But training from scratch is giving me loww accuracy i am forced to keep the batch size 8 as i have single GPU.

Thanks

dbash commented 1 year ago

Keeping the batch size large is important for the overall segmentation quality. I suggest using the same batch size (or larger) as in our experiments to reproduce the results. If your computational resources are limited you can use some techniques such as gradient accumulation.