How to inference on single image and draw the bbox on the image?

fredzzhang / pvic

[ICCV'23] Official PyTorch implementation for paper "Exploring Predicate Visual Context in Detecting Human-Object Interactions"

BSD 3-Clause "New" or "Revised" License

68 stars 8 forks source link

How to inference on single image and draw the bbox on the image? #30

Closed aixiaodewugege closed 1 year ago

aixiaodewugege commented 1 year ago

Thank you for your kind words! I'd like to evaluate the model on my dataset and visualize the outcomes. How can I proceed with this?

fredzzhang commented 1 year ago

Hi @aixiaodewugege,

The inference code is in the inference.py script, but it hasn't been cleaned up yet. I'm currently occupied by a few other things and will clean up the repo in September.

Fred.

fredzzhang commented 1 year ago

Hi @aixiaodewugege,

The codebase for inference and visualisation has now been updated.

Cheers, Fred.