How to visualize the learned groups？

[YwSama]

Thank you for your great work!
I'm curious about the visualization of the learned groups in 5th and 10th columns of Figure 3, May I ask if there is any code for generating them? Thanks!

[Jazzcharles]

You can try to specify --vis argument in main_seg.py