Thank you for your excellent work! I have read through your paper, and I'm curious about the visualization of the mask in the Appendix Fig. 1. Seems that there is no corresponding visualization option in https://github.com/NVlabs/GroupViT/blob/506ba5f8c33ed947ca122f70e0543c12597d1040/segmentation/evaluation/group_vit_seg.py. May I ask if there is any code for generating the mask in style of Appendix Fig. 1? Thanks!