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

fredzzhang / pvic

Official PyTorch implementation for ICCV2023 paper "Exploring Predicate Visual Context in Detecting Human-Object Interactions"

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

56 stars 7 forks source link

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

Closed aixiaodewugege closed 8 months ago

aixiaodewugege commented 10 months 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 10 months 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 8 months ago

Hi @aixiaodewugege,

The codebase for inference and visualisation has now been updated.

Cheers, Fred.