isl-org / lang-seg

Language-Driven Semantic Segmentation
MIT License
691 stars 84 forks source link

Training set used in demo model #39

Open loris2222 opened 1 year ago

loris2222 commented 1 year ago

Good day,

I was wondering whether the demo model available from the repo (demo_e200.ckpt) was solely trained on ADE20K as specified in the repo or whether it was trained on all 7 datasets presented in section 5.2. This is unclear to me since the demo model works well with classes that are not covered by ADE20K such as the animals, and these classes are covered by others such as COCO.

In case it was trained on multiple datasets, I would like to know how to do so myself.

Thank you in advance.

robin-karlsson0 commented 11 months ago

Found this comment indicating that the demo model is trained on 7 concatenated datasets, NOT just on ADE20K.

https://github.com/isl-org/lang-seg/issues/13#issuecomment-1095814404