Skip to content

Latest commit

 

History

History
37 lines (26 loc) · 788 Bytes

README.md

File metadata and controls

37 lines (26 loc) · 788 Bytes
Raw

PIC_HOIW

This repository provides python API that assists in loading, parsing and visualizing data in HOIW.

Install

  1. Clone this repo:

    git clone https://github.com/YueLiao/PIC_HOIW.git

  2. Install the requirements

    pip install -r requirements.txt

Use API

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')