UniHSI

[ICLR 2024 Spotlight] Unified Human-Scene Interaction via Prompted Chain-of-Contacts

Generate Convert Improve

Install / Use

npx skills add InternRobotics/UniHSI

Installs into whichever agent you are using.

About this skill

Quality Score

0/100

README

<br> <p align="center"> <h1 align="center"><img src="assets/unihsi_icon.png" align="center" width="6.5%"><strong>Unified Human-Scene Interaction via Prompted Chain-of-Contacts</strong></h1> <p align="center"> <a href='https://github.com/xizaoqu/' target='_blank'>Zeqi Xiao</a>&emsp; <a href='https://tai-wang.github.io/' target='_blank'>Tai Wang</a>&emsp; <a href='https://scholar.google.com/citations?user=GStTsxAAAAAJ&hl=zh-CN&oi=ao/' target='_blank'>Jingbo Wang</a>&emsp; <a href='https://www.jinkuncao.com/' target='_blank'>Jinkun Cao</a>&emsp; <a href='http://zhangwenwei.cn/' target='_blank'>Wenwei Zhang</a>&emsp; <a href='https://daibo.info/' target='_blank'>Bo Dai</a>&emsp; <a href='http://dahua.site/' target='_blank'>Dahua Lin</a>&emsp; <a href='https://oceanpang.github.io/' target='_blank'>Jiangmiao Pang*</a>&emsp; <br> Shanghai AI Laboratory&emsp;Nanyang Technological University&emsp;Carnegie Mellon University </p> </p> <p align="center"> <a href="https://arxiv.org/abs/2309.07918" target='_blank'> <img src="https://img.shields.io/badge/arXiv-2309.07918-blue?"> </a> <a href="./assets/UniHSI.pdf" target='_blank'> <img src="https://img.shields.io/badge/Paper-📖-blue?"> </a> <a href="https://xizaoqu.github.io/unihsi" target='_blank'> <img src="https://img.shields.io/badge/Project-&#x1F680-blue"> </a> </p>

🏠 About

<div style="text-align: center;"> <img src="assets/teaser.png" alt="Dialogue_Teaser" width=100% > </div> This paper presents <b>a UNIfied HSI framework, UniHSI, which supports unified control of diverse interactions through language commands</b>. This framework is built upon the definition of interaction as <b>Chain of Contacts</b> (CoC): steps of human joint-object part pairs, which is inspired by the strong correlation between interaction types and human-object contact regions. Based on the definition, UniHSI constitutes a <b>Large Language Model (LLM) Planner</b> to translate language prompts into task plans in the form of CoC, and a <b>Unified Controller</b> that turns CoC into uniform task execution. To facilitate training and evaluation, we collect a new dataset named <b>ScenePlan</b> that encompasses thousands of task plans generated by LLMs based on diverse scenarios. Comprehensive experiments demonstrate the effectiveness of our framework in versatile task execution and generalizability to real scanned scenes.

🔥 News

For reproducing the demo, please refer to blender_for_UniHSI
[2024-04] The data is released.
[2024-03] The code is released.
[2024-01] UniHSI is accepted as ICLR 2024 spotlight. Thanks for the recognition!
[2023-09] We release the paper of UniHSI. Please check the :point_right: webpage :point_left: and view our demos! :sparkler:;

🔍 Overview

<p align="center"> <img src="assets/main_figure.jpg" align="center" width="100%"> </p> The whole pipeline consists of two major components: the LLM Planner and the Unified Controller. The LLM planner takes language inputs and background scenario information as inputs and outputs multi-step plan in the form of a Chain of Contacts. The Unified Controller then executes task plans step-by-step and output interaction movements.

Installation

Download Isaac Gym from the website, then follow the installation instructions.

Once Isaac Gym is installed, install the external dependencies for this repo:

pip install -r requirement.txt

Data Preparation

PartNet

Download PartNet and ShapeNet V2.
Save them in the following formation

data/
├── partnet_origin
│   ├── obj_id1
│   ├── obj_id2
│   ├── ...
├── shapenet_origin
│   ├── class_id1
│   │    ├── obj_id1
│   │    ├── ...
│   ├── class_id2
│   │    ├── obj_id1
│   │    ├── ...
│   ├── ...

Extract the objects used in sceneplan by

python cp_partnet_train.py
python cp_partnet_test.py

ScanNet

Download ScanNet.
Save them in the following formation

data/
├── scan_origin
│   ├── scans
│   │   ├── scans_1
│   │   ├── scans_2
│   │   ├── ...

Extract the objects used in sceneplan by

python cp_scannet_test.py

Motio Clips

We select and process motion clips from SAMP and CIRCLE.

Training

We adopt step-by-step training.

sh train_partnet_simple.sh
sh train_partnet_mid.sh
sh train_partnet_hard.sh

Demo

sh demo_scannet.sh

Evaluation

sh test_partnet_simple.sh
sh test_partnet_mid.sh
sh test_partnet_hard.sh
sh test_scannet_simple.sh
sh test_scannet_mid.sh
sh test_scannet_hard.sh

<table> <tr> <td rowspan="2">Source</td> <td colspan="3">Success Rate (%) </td> <td colspan="3">Contact Error</td> <td colspan="3">Success Steps</td> </tr > <tr > <td>Simple</td> <td>Mid</td> <td>Hard</td> <td>Simple</td> <td>Mid</td> <td>Hard</td> <td>Simple</td> <td>Mid</td> <td>Hard</td> </tr> <tr > <td>PartNet</td> <td>85.5</td> <td>67.9</td> <td>40.5</td> <td>0.035</td> <td>0.037</td> <td>0.040</td> <td>2.13</td> <td>4.11</td> <td>4.84</td> </tr> <tr > <td>ScanNet</td> <td>73.2</td> <td>43.1</td> <td>22.3</td> <td>0.061</td> <td>0.072</td> <td>0.062</td> <td>2.21</td> <td>3.47</td> <td>4.78</td> </tr> </table>

The results will be saved in the "output" folder.

There will be ~10% variance due to randomness in sampling.

🔗 Citation

If you find our work helpful, please cite:

@inproceedings{
  xiao2024unified,
  title={Unified Human-Scene Interaction via Prompted Chain-of-Contacts},
  author={Zeqi Xiao and Tai Wang and Jingbo Wang and Jinkun Cao and Wenwei Zhang and Bo Dai and Dahua Lin and Jiangmiao Pang},
  booktitle={The Twelfth International Conference on Learning Representations},
  year={2024},
  url={https://openreview.net/forum?id=1vCnDyQkjg}
}

📄 License

<a rel="license" href="http://creativecommons.org/licenses/by-nc-sa/4.0/"><img alt="Creative Commons License" style="border-width:0" src="https://i.creativecommons.org/l/by-nc-sa/4.0/80x15.png" /></a> <br /> This work is under the <a rel="license" href="http://creativecommons.org/licenses/by-nc-sa/4.0/">Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License</a>.