How to test the custom images? / Visualize the generated scene graph for the custom images

liuhengyue / fcsgg

A PyTorch implementation for the paper: Fully Convolutional Scene Graph Generation, CVPR 2021

MIT License

27 stars 2 forks source link

How to test the custom images? / Visualize the generated scene graph for the custom images #4

Closed dyang-TUM closed 10 months ago

dyang-TUM commented 2 years ago

Hi, @liuhengyue could you please provide the way to test the custom images and visualize the generated scene graph after training? It seems like I need to use visualize_output.py.

Thanks for your help!

liuhengyue commented 2 years ago

For testing images from the dataloader, you can use the script tools/visualize_output.py, and tools/custom_predict.py for custom images.