This repository provides python API that assists in loading, parsing and visualizing data in HOIW.
-
Clone this repo:
git clone https://github.com/YueLiao/PIC_HOIW.git
-
Install the requirements
pip install -r requirements.txt
For HOI visulization on an image, run:
python demo.py --image_dir ./demo/images --annot_file ./demo/example.json --image_name trainval_000001.png
For HOI visulization on all images in an annotation file, run:
python demo.py --image_dir ./demo/images --annot_file ./demo/example.json --output_path vis_hoiw
For evaluating your result, run:
from evalution import hoiw_eval
eval_demo = hoiw_eval('annotation.json')
eval_demo.evalution('prediction.json')