Introduction

This repo is used to implement HOI Detection method.

Update

• Browse the HICO-Det with the ./tools/mics/browse_dataset.py
• Reimplement QPIC (CVPR 2021).
• Image demo (Inference a single image and visualization).
• Evaluation.

Installation

This codebase is based on mmdetection(v2.16.0). Please refer to INSTALL.md for installation and dataset preparation.

Usage

Browse the dataset

python tools/misc/browse_dataset.py configs/qpic/qpic_r50_150e_hico.py --output-dir ./results/hico-det

Inference demo

Note: The pretrained config can be download. Baidu:qpic

python demo/image_demo.py ./demo/demo.jpg configs/qpic/qpic_r50_150e_hico.py ./checkpoints/qpic_r50_150e_hico-adf11cf1.pth

Train the model

python tools/train.py configs/qpic/qpic_r50_150e_hico.py # train with 1 gpu
tools/dist_train.sh configs/qpic/qpic_r50_150e_hico.py 4  # distributed training with 4 gpus

Note

• To avoid re-registry a same name module in other OpenMMlab project, MMHOIDet renames the registry like PIPELINES->HOI_PIPELINES, DATASETS->HOI_DATASETS

Contribuation

Any form help will be welcomed. Feel free to post an issue or PR to contribute to this codebase :)