Add segmentation?

[straeter]

Hi, nice work! Any reason why you have not (yet) implemented the image segmentation? Maybe because it is unclear how they implemented it exactly or any troubles with that or just no time for that? Also, what FID have you achieved? Thanks!

[confifu]

The PCSegRGB-Sep-Disc branch has some changes that handle seg maps. The exact reason is a combination of all the reasons you mentioned. :) The description was not clear so I tried out some ways but they didn't work that well. If you want to look into it there is this notebook that I used for training this variation. Also I used this notebook for generating human seg maps. I think the best FID score was ~10.