Skip to content

Latest commit

 

History

History
240 lines (164 loc) · 9.83 KB

README.md

File metadata and controls

240 lines (164 loc) · 9.83 KB

HUMANISE: Language-conditioned Human Motion Generation in 3D Scenes

Paper arXiv Paper PDF Project Page Dataset

Zan Wang, Yixin Chen, Tengyu Liu, Yixin Zhu, Wei Liang, Siyuan Huang

This repository is an official implementation of "HUMANISE: Language-conditioned Human Motion Generation in 3D Scenes".

In this work, we propose a large-scale and semantic-rich human-scene interaction dataset, HUMANISE. It has language description for each human-scene interaction. HUMANISE enables a new task: language-conditioned human motion generation in 3D scenes.

Paper | arXiv | Project Page | Data

Update

  • 2023/11/22: We plan to refactor this repository to optimize the framework and code in recent weeks.
  • 2023/6/5: We re-format the data for using HUMANISE easier. Please check HUMANISE New Format for more details.

Abstract

Learning to generate diverse scene-aware and goal-oriented human motions in 3D scenes remains challenging due to the mediocre characteristics of the existing datasets on Human-Scene Interaction(HSI); they only have limited scale/quality and lack semantics. To fill in the gap, we propose a large-scale and semantic-rich synthetic HSI dataset, denoted as HUMANISE, by aligning the captured human motion sequences with various 3D indoor scenes. We automatically annotate the aligned motions with language descriptions that depict the action and the unique interacting objects in the scene; e.g., sit on the armchair near the desk. HUMANISE thus enables a new generation task, language-conditioned human motion generation in 3D scenes. The proposed task is challenging as it requires joint modeling of the 3D scene, human motion, and natural language. To tackle this task, we present a novel scene-and-language conditioned generative model that can produce 3D human motions of the desirable action interacting with the specified objects. Our experiments demonstrate that our model generates diverse and semantically consistent human motions in 3D scenes.

Preparation

1. Environment Setup

  • Install the following key libraries:

    • pytorch, numpy, trimesh, smplx, pyquaternion, pyrender, scikit-learn, natsort, pillow, tqdm, ...
  • Make sure your machine supports on-screen/off-screen rendering.

Notes: we run our code with pytorch 1.10 and cuda11.3.

2. Data Preparation

  1. ScanNet V2 Dataset

    Remember to change the dataset folder configuration in utils/configuration.py.

  2. Our pre-synthesized data, or you can generate your own data with our pipeline, see HUMANISE Synthesis for more details.

  3. SMPLX v1.1

HUMANISE Dataset

1. Synthesis

See HUMANISE Synthesis for more details.

2. Visualization

For HUMANISE dataset visualization, we provide rendering script visualize_dataset.py which will render an animation video with top-down view. The result will be saved in ./tmp/.

  • on-screen rendering
python visualize_dataset.py --pkl ${PKL} --index ${index} --vis
# python visualize_dataset.py --pkl your_path/lie/scene0000_001810_c71dc702-1f1d-4381-895c-f07e9a10876b/anno.pkl --index 0 --vis

Notes: --vis will render the static human-scene interaction with trimesh on screen.

  • off-screen rendering
PYOPENGL_PLATFORM=egl python visualize_dataset.py --pkl ${PKL} --index ${index}
# PYOPENGL_PLATFORM=egl python visualize_dataset.py --pkl your_path/lie/scene0000_001810_c71dc702-1f1d-4381-895c-f07e9a10876b/anno.pkl --index 0

See more information about the data format.

Our Model

Preprocess ScanNet Scenes

Following link to preprocess the ScanNet scenes; then change the preprocess_scene_folder configuration in utils/configuration.py.

Action-Specific Model

  • Train

    bash scripts/train.sh "${ACTION}"
    # e.g., bash scripts/train.sh "walk"
  • Eval (Quantitative)

    • Reconstruction metrics.

      bash scripts/eval_metric.sh ${STAMP} "${ACTION}"
      # e.g., bash scripts/eval_metric.sh 20220829_194320 "walk"
    • Generation metrics (goal dist.). Set eval_rec=False (Line 64) in eval_metric_motion.py to compute generation metrics. Then run the following script. It will take several hours to compute the results.

      bash scripts/eval_metric.sh ${STAMP} "${ACTION}"
      # e.g., bash scripts/eval_metric.sh 20220829_194320 "walk"
    • Generation metrics (APD).

      bash scripts/eval_pairwise_distance.sh ${STAMP} "${ACTION}"
      # e.g., bash scripts/eval_pairwise_distance.sh 20220829_194320 "walk"
  • Eval (Qualitative)

    • Qualitative results of reconstruction.

      bash scripts/eval.sh ${STAMP} "${ACTION}"
      # e.g., bash scripts/eval.sh 20220829_194320 "walk"
    • Qualitative results of generation. Comment Line 36 and uncomment Line 35 in eval_motion.py. The parameter k in solver.save_k_sample(k: int) indicates the number of samples for each case.

      bash scripts/eval.sh ${STAMP} "${ACTION}"
      # e.g., bash scripts/eval.sh 20220829_194320 "walk"

Action-Agnostic Model

  • First, change to agnostic branch by executing git checkout agnostic.

  • Train

    bash scripts/train.sh
  • Eval (Quantitative)

    • Reconstruction metrics.

      bash scripts/eval_metric.sh ${STAMP}
      # e.g., bash scripts/eval_metric.sh 20220831_153356
    • Generation metrics (goal dist.). Set eval_rec=False (Line 64) in eval_metric_motion.py to compute generation metrics. Then run the following script. It will take several hours to compute the results.

      bash scripts/eval_metric.sh ${STAMP}
      # e.g., bash scripts/eval_metric.sh 20220831_153356
    • Generation metrics (APD).

      bash scripts/eval_pairwise_distance.sh ${STAMP}
      # e.g., bash scripts/eval_pairwise_distance.sh 20220831_153356
  • Eval (Qualitative)

    • Qualitative results of generation.

      bash scripts/eval.sh ${STAMP}
      # e.g., bash scripts/eval.sh 20220831_153356

Pretrained Models

You can use our pretrained models. (In the checkpoints folder)

STAMP Pretrained Models
POINTTRANS_C_32768 scene model (point transformer)
20220829_194320 action-specific model (walk)
20220830_203617 action-specific model (sit)
20220830_203832 action-specific model (stand up)
20220830_204043 action-specific model (lie)
20220831_153356 action-agnostic model

Put the downloaded checkpoints into outputs/ folder as following:

-| model/
-| outputs/
---| POINTTRANS_C_32768/
---| 20220829_194320/
---| ...
-| scripts/
-| ...

Citation

If you find our project useful, please consider citing us:

@inproceedings{wang2022humanise,
  title={HUMANISE: Language-conditioned Human Motion Generation in 3D Scenes},
  author={Wang, Zan and Chen, Yixin and Liu, Tengyu and Zhu, Yixin and Liang, Wei and Huang, Siyuan},
  booktitle={Advances in Neural Information Processing Systems (NeurIPS)},
  year={2022}
}

Acknowledgements

Some codes are borrowed from PSI-release, point-transformer, Pointnet2.ScanNet, and YouRefIt_ERU.

License

Our code and data are released under the MIT license. The following datasets are used in our project and are subject to their respective licenses: