Windsander / HAKE-Action-Torch

HAKE-Action in PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

HAKE-Action-Torch

Eight-in-One: CVPR'18 (Part States), CVPR'19 (interactiveness), CVPR'20 (PaStaNet, Dj-RN, SymNet), NeurIPS'20 (IDN), TPAMI(Upgraded TIN, Upgraded SymNet).

HAKE-Action-Torch (PyTorch) is a project to open the SOTA action understanding studies based on our project: Human Activity Knowledge Engine. It includes SOTA models and their corresponding HAKE-enhanced versions based on our related papers (CVPR'18/19/20, NeurIPS'20, PAMI'21). The TensorFlow version of HAKE-Action is here.

Currently, it is manintained by Yong-Lu Li, Xinpeng Liu, and Hongwei Fan.

News: (2022.02.14) We release the human body part state labels based on AVA: HAKE-AVA and HAKE 2.0 paper.

(2021.10.06) Our extended version of SymNet is accepted by TPAMI! Paper and code are coming soon.

(2021.2.7) Upgraded HAKE-Activity2Vec is released! Images/Videos --> human box + ID + skeleton + part states + action + representation. [Description]

Full demo: [YouTube], [bilibili]

(2021.1.15) Our extended version of TIN (Transferable Interactiveness Network) is accepted by TPAMI!

(2020.10.27) The code of IDN (Paper) in NeurIPS'20 is released!

Project

HAKE-Action-Torch
  ├──Master Branch                          # Unified pipeline; CVPR'18/20, PaStanet and Part States.
  ├──IDN-(Integrating-Decomposing-Network)  # NeurIPS'20, HOI Analysis: Integrating and Decomposing Human-Object Interaction.
  ├──DJ-RN-Torch                            # CVPR'20, Detailed 2D-3D Joint Representation for Human-Object Interaction.
  ├──TIN-Torch                              # CVPR'19, Transferable Interactiveness Knowledge for Human-Object Interaction Detection.
  └──SymNet-Torch                           # CVPR'20, Symmetry and Group in Attribute-Object Compositions.

Papers

Results on HICO-DET with different object detections.

Method Detector HAKE Full(def) Rare(def) None-Rare(def) Full(ko) Rare(ko) None-Rare(ko)
TIN COCO - 17.54 13.80 18.65 19.75 15.70 20.96
TIN COCO HAKE-HICO-DET 22.12 20.19 22.69 24.06 22.19 24.62
TIN COCO HAKE-Large 22.66 21.17 23.09 24.53 23.00 24.99
TIN-PAMI COCO - 20.93 18.95 21.32 23.02 20.96 23.42
DJ-RN COCO - 21.34 18.53 22.18 23.69 20.64 24.60
IDN COCO - 23.36 22.47 23.63 26.43 25.01 26.85
IDN COCO+HICO-DET - 26.29 22.61 27.39 28.24 24.47 29.37
TIN GT Pairs - 34.26 22.90 37.65 - - -
IDN GT Pairs - 43.98 40.27 45.09 - - -

Results on V-COCO.

As VCOCO is built on COCO, thus finetuning detector on VCOCO basically contributes marhinally to performance.

Method HAKE AP(role)
TIN - 47.8
TIN HAKE-Large 51.0
TIN-PAMI - 49.1
IDN - 53.3

Results on Ambiguous-HOI.

Method mAP
TIN 8.22
DJ-RN 10.37

Results on PaStaNet-HOI

Method mAP
TIN-PAMI 15.38

Modules

Activity2Vec (CVPR'20)

The independent Torch version is in: Activity2Vec (A2V).

IDN (NeurIPS'20)

The independent Torch version is in: IDN.

DJ-RN (CVPR'20)

The independent Torch version is in: DJ-RN-Torch

TIN (CVPR'19)

The independent Torch version is in: TIN-Torch

SymNet (CVPR'20)

Coming soon.

Citation

If you find our works useful, please consider citing:

---HAKE 2.0
@misc{li2022hake,
  title={HAKE: A Knowledge Engine Foundation for Human Activity Understanding}, 
  author={Yong-Lu Li and Xinpeng Liu and Xiaoqian Wu and Yizhuo Li and Zuoyu Qiu and Liang Xu and Yue Xu and Hao-Shu Fang and Cewu Lu},
  year={2022},
  eprint={2202.06851},
  archivePrefix={arXiv},
  primaryClass={cs.CV}
}
---SymNet-PAMI
@article{li2021learning,
  title={Learning Single/Multi-Attribute of Object with Symmetry and Group},
  author={Li, Yong-Lu and Xu, Yue and Xu, Xinyu and Mao, Xiaohan and Lu, Cewu},
  journal={TPAMI},
  year={2021}
}
---TIN-PAMI
@article{li2021transferable,
  title={Transferable Interactiveness Knowledge for Human-Object Interaction Detection},
  author={Li, Yong-Lu and Liu, Xinpeng and Wu, Xiaoqian and Huang, Xijie and Xu, Liang and Lu, Cewu},
  journal={TPAMI},
  year={2021}
}
---IDN:
@inproceedings{li2020hoi,
  title={HOI Analysis: Integrating and Decomposing Human-Object Interaction},
  author={Li, Yong-Lu and Liu, Xinpeng and Wu, Xiaoqian and Li, Yizhuo and Lu, Cewu},
  booktitle={NeurIPS},
  year={2020}
}
---HAKE:
@inproceedings{li2020pastanet,
  title={PaStaNet: Toward Human Activity Knowledge Engine},
  author={Li, Yong-Lu and Xu, Liang and Liu, Xinpeng and Huang, Xijie and Xu, Yue and Wang, Shiyi and Fang, Hao-Shu and Ma, Ze and Chen, Mingyang and Lu, Cewu},
  booktitle={CVPR},
  year={2020}
}
@inproceedings{lu2018beyond,
  title={Beyond holistic object recognition: Enriching image understanding with part states},
  author={Lu, Cewu and Su, Hao and Li, Yonglu and Lu, Yongyi and Yi, Li and Tang, Chi-Keung and Guibas, Leonidas J},
  booktitle={CVPR},
  year={2018}
}
---DJ-RN
@inproceedings{li2020detailed,
  title={Detailed 2D-3D Joint Representation for Human-Object Interaction},
  author={Li, Yong-Lu and Liu, Xinpeng and Lu, Han and Wang, Shiyi and Liu, Junqi and Li, Jiefeng and Lu, Cewu},
  booktitle={CVPR},
  year={2020}
}
---TIN
@inproceedings{li2019transferable,
  title={Transferable Interactiveness Knowledge for Human-Object Interaction Detection},
  author={Li, Yong-Lu and Zhou, Siyuan and Huang, Xijie and Xu, Liang and Ma, Ze and Fang, Hao-Shu and Wang, Yanfeng and Lu, Cewu},
  booktitle={CVPR},
  year={2019}
}
---SymNet
@inproceedings{li2020symmetry,
  title={Symmetry and Group in Attribute-Object Compositions},
  author={Li, Yong-Lu and Xu, Yue and Mao, Xiaohan and Lu, Cewu},
  booktitle={CVPR},
  year={2020}
}

TODO

  • Refined Activity2Vec

HAKE

HAKE[website] is a new large-scale knowledge base and engine for human activity understanding. HAKE provides elaborate and abundant body part state labels for active human instances in a large scale of images and videos. With HAKE, we boost the action understanding performance on widely-used human activity benchmarks. Now we are still enlarging and enriching it, and looking forward to working with outstanding researchers around the world on its applications and further improvements. If you have any pieces of advice or interests, please feel free to contact Yong-Lu Li (yonglu_li@sjtu.edu.cn).

If you get any problems or if you find any bugs, don't hesitate to comment on GitHub or make a pull request!

HAKE-Action-Torch is freely available for free non-commercial use, and may be redistributed under these conditions. For commercial queries, please drop an e-mail. We will send the detail agreement to you.

About

HAKE-Action in PyTorch

License:Apache License 2.0