Training code / weight for COCO

[1211sh]

Hi,
could you please provide the code for training COCO (e.g., dataloader, segmenter config) or weight file of COCO?

Thank you.

[YumengLi007]

Hi @1211sh , thanks for your interests!
We have not actively experimented with COCO dataset. I can work on collecting code related to COCO and go through the approval process of the open sourcing (e.g., code scan).
Meanwhile, maybe you could try out the checkpoint trained on ADE20K? Thank you :)

[1211sh]

Thanks a lot for your quick reply and sincere help.

I have already tested on Ade20K, and I think it would be even more meaningful for the research community if possible with the COCO dataset, given the good results of your research. :)

[YumengLi007]

Hi @1211sh I have uploaded the checkpoint here, and updated the code and README file, please have a look :) You should be able to generate images using bash_script/generate_coco.sh. Please make sure converting the COCO labels firstly using convert_coco_stuff_id.ipynb.

[1211sh]

Hi, I have found out the code with the new checkpoint is working well. Thanks a lot!

[1211sh]

Dear sir,
I found that 'decoder_epoch_50_172cls.pth' is missing in https://huggingface.co/Yumeng/ALDM/tree/main.
Could you please provide the weight?

Thank you.

[YumengLi007]

Hi @1211sh , please find the model here