Questions about the post-processing for keypoint detection and segmentation

[essunny310]

Hi, very interesting work!

As mentioned in the paper, a lightweight U-Net is leveraged to post-processes the output image to generate some heatmaps (for keypoint detection) or masks (for segmentation). Which specific model did you use? How did you train the U-Net? It would be helpful if you could provide more details about this.

Thank you.

[Gengzigang]

Extracting keypoints or masks from the images generated by our editor is a simple mapping process, so the UNet will have minimal impact on performance.

[essunny310]

Thanks for your reply.

Could you please provide more details about this procedure? Thanks a lot.