meng-tang / rloss

Regularized Losses (rloss) for Weakly-supervised CNN Segmentation
MIT License
207 stars 47 forks source link

mIoU is only 0.579 much lower than your reported 0.69 #13

Open l44xu opened 4 years ago

l44xu commented 4 years ago

Hi Meng,

I tried the pytorch code using exactly the instructed steps. With 60 epochs, I only got mIOU of 0.579, much lower than your reported 0.69. Is there something that I should do to improve the accuracy?

I used the below command:

python train_withdensecrfloss.py --backbone mobilenet --lr 0.007 --workers 6 --epochs 60 --batch-size 8 --checkname deeplab-mobilenet --eval-interval 2 --dataset pascal --save-interval 2 --densecrfloss 2e-9 --rloss-scale 0.5 --sigma-rgb 15 --sigma-xy 100

Many thanks!

Best, Linlin

meng-tang commented 4 years ago

For debugging, could you try to train without any regularization, just the partial cross-entropy loss? I would expect 65.x% in this case.