How to use the VG pretrained model for visualization in the Gradio Webui Demo

[RunpuWei]

I successfully tested the demo use the pre-trained model on COCO that you published, but CUDA errors popped up when I used the pre-trained model on VG that you published. How should I solve this problem, thank you!

[ZGCTroy]

I think the reason is that VG and COCO have different class categories. The visualization code is only a demo specifically designed for COCO-stuff. So you may need to do some modifications.