Training code / weight for COCO about aldm HOT 6 OPEN

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. :)

from aldm.

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 :)

from aldm.

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.

from aldm.

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

from aldm.

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.

from aldm.

Hi @1211sh , please find the model here

from aldm.