Skip to content

Latest commit

 

History

History
48 lines (31 loc) · 1.41 KB

README.md

File metadata and controls

48 lines (31 loc) · 1.41 KB

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