stragen generation with provided checkpoints

[JialeTao]

stragen generation with provided checkpoints

[xingzhehe]

Thanks for bringing this up. It took me quite some time to fix. Please check the latest version.

I believe this is a bug caused by my code cleaning. BTW, even now, I don't know why it causes a bug since it is just a different writing way. You can check the self.gen_keypoints_embedding_noise, self.gen_keypoints_layer and self.gen_background_embedding in generator.py.

If you got any further problems, don't hesitate to contact me.

[JialeTao]

Thanks for quick reply! After modify these three components, it works now.

It seems the generated images are much more natural now, but there still exists many artifacts, and some bodies are deformed. Is this normal? Since some parts are missed, does it hurt the performance on the segmentation?

[xingzhehe]

I unrolled all the modules to hard-coded ones (please update). It works better now. I doubt it is a bug in Pytorch, not sure if it is due to the version.

There are still artifacts, which is normal since I do not use any path regularization or EMA or EqualLinear for simplifity. It may harm the quality of segmentation, but we just found the current one is enough for a SOTA performance.