ttanida / rgrg

Code for the CVPR paper "Interactive and Explainable Region-guided Radiology Report Generation"
MIT License
128 stars 23 forks source link

Visual Grounding code #26

Open lvchw opened 5 months ago

lvchw commented 5 months ago

Hi, thank you for your wonderful work!

I want to perform Visual Grounding on the Chest ImaGnome to visualize the results of report generation, but I failed to found any works to perform VG task on this dataset. I noticed that, in your paper, Appendix B.2 presents three excellent VG results. How are they produced? Is the code in the repository? If not, is it available? I would greatly appreciate it if you could help me.

ttanida commented 5 months ago

Hi,

I am afraid you will have to write some custom code for this, as I didn't commit the code back then.

You can adjust this script by e.g. adjusting this line such that the model also outputs the bounding boxes together besides the beam search output. Then it's only a matter of the plotting the color coded region sentences and associated bounding boxes together.