WXinlong / DenseCL

Dense Contrastive Learning (DenseCL) for self-supervised representation learning, CVPR 2021 Oral.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Semantic segmentation on PASCAL VOC

MaitaYuki opened this issue · comments

@WXinlong,

Thanks for sharing your great work!
I was able to reproduce your object detection result on Pascal VOC.
However, when I tested semantic segmentation on Pascal VOC using your pre-trained model on ImageNet1k "densecl_r50_imagenet_200ep.pth", I got mIoU 0.62, which is worse than the 0.69 reported in your paper. My test procedure is explained below,

  1. Install your modified mmsegmentation
  2. Download "densecl_r50_imagenet_200ep.pth" from your website
  3. Update the 5th line of code in fcn_r50-d8.py to pretrained='/pretrained/densecl_r50_imagenet_200ep.pth'
  4. Run ./tools/dist_train.sh configs/densecl/fcn_r50-d8_512x512_20k_voc12aug.py 2 --work-dir models/fcn_r50-d8_512x512_20k_voc12aug (running on 2 GPUs)

I got a result of mIoU 0.62, mAcc: 0.75, aAcc: 0.91 at the end of the training. I ran 3 rounds and got similar results. Attached are my configuration file and training log.

Do you know a possible reason?
Thanks!

20230213_204945.log
fcn_r50-d8_512x512_20k_voc12aug.zip