IDEA-Research / MotionLLM

[Arxiv-2024] MotionLLM: Understanding Human Behaviors from Human Motions and Videos
https://lhchen.top/MotionLLM
Other
180 stars 4 forks source link

MotionLLM: Understanding Human Behaviors from Human Motions and Videos

task

Ling-Hao Chen😎 1, 3, Shunlin Lu😎 2, 3, Ailing Zeng3, Hao Zhang3, 4, Benyou Wang2, Ruimao Zhang2, Lei ZhangπŸ€— 3

😎Co-first author. Listing order is random. πŸ€—Corresponding author.

1Tsinghua University, 2School of Data Science, The Chinese University of Hong Kong, Shenzhen (CUHK-SZ), 3International Digital Economy Academy (IDEA), 4The Hong Kong University of Science and Technology

πŸ“° News

🀩 Abstract

This study delves into the realm of multi-modality (i.e., video and motion modalities) human behavior understanding by leveraging the powerful capabilities of Large Language Models (LLMs). Diverging from recent LLMs designed for video-only or motion-only understanding, we argue that understanding human behavior necessitates joint modeling from both videos and motion sequences (e.g., SMPL sequences) to capture nuanced body part dynamics and semantics effectively. In light of this, we present MotionLLM, a straightforward yet effective framework for human motion understanding, captioning, and reasoning. Specifically, MotionLLM adopts a unified video-motion training strategy that leverages the complementary advantages of existing coarse video-text data and fine-grained motion-text data to glean rich spatial-temporal insights. Furthermore, we collect a substantial dataset, MoVid, comprising diverse videos, motions, captions, and instructions. Additionally, we propose the MoVid-Bench, with carefully manual annotations, for better evaluation of human behavior understanding on video and motion. Extensive experiments show the superiority of MotionLLM in the caption, spatial-temporal comprehension, and reasoning ability.

🀩 Highlight Applications

application

πŸ”§ Technical Solution

system

πŸ’» Try it

We provide a simple online demo for you to try MotionLLM. Below is the guidance to deploy the demo on your local machine.

Step 1: Set up the environment

pip install -r requirements.txt

Step 2: Download the pre-trained model

2.1 Download the LLM Please follow the instruction of [Lit-GPT](https://github.com/Lightning-AI/litgpt) to prepare the LLM model (vicuna 1.5-7B). These files will be: ```bah ./checkpoints/vicuna-7b-v1.5 β”œβ”€β”€ generation_config.json β”œβ”€β”€ lit_config.json β”œβ”€β”€ lit_model.pth β”œβ”€β”€ pytorch_model-00001-of-00002.bin β”œβ”€β”€ pytorch_model-00002-of-00002.bin β”œβ”€β”€ pytorch_model.bin.index.json β”œβ”€β”€ tokenizer_config.json └── tokenizer.model ``` If you have any confusion, we will update a more detailed instruction in couple of days.
2.2 Dowload the LoRA and the projection layer of the MotionLLM We now release one versions of the MotionLLM checkpoints, namely `v1.0` (download [here](https://drive.google.com/drive/folders/1d_5vaL34Hs2z9ACcMXyPEfZNyMs36xKx?usp=sharing)). Opening for the suggestions to Ling-Hao Chen and Shunlin Lu. ```bash wget xxx ``` Keep them in a folder named and remember the path (`LINEAR_V` and `LORA`).

2.3 Run the demo

Choice 1: gradio demo ```bash GRADIO_TEMP_DIR=temp python app.py --lora_path $LORA --mlp_path $LINEAR_V ``` If you have some error in downloading the huggingface model, you can try the following command with the mirror of huggingface. ```bash HF_ENDPOINT=https://hf-mirror.com GRADIO_TEMP_DIR=temp python app.py --lora_path $LORA --mlp_path $LINEAR_V ``` The `GRADIO_TEMP_DIR=temp` defines a temporary directory as `./temp` for the Gradio to store the data. You can change it to your own path. After thiess, you can open the browser and visit the local host via the command line output reminder. If it is not loaded, please change the IP address as your local IP address (via command `ifconfig`).
Choice 2: CLI demo We also provide a CLI demo for you to try the MotionLLM. You can run the following command to try the MotionLLM. ```bash python cli.py --lora_path $LORA --mlp_path $LINEAR_V ``` During inference, you can input the video path and your question to get the answer. ```bash # Example here Input video path: xxx.mp4 Your question: what xxx ? ================================ The man plan to xxx. ================================ ```

πŸ’Ό To-Do

πŸ’‹ Acknowledgement

The author team would like to deliver many thanks to many people. Qing Jiang helps a lot with some parts of manual annotation on MoVid Bench and resolves some ethics issues of MotionLLM. Jingcheng Hu provided some technical suggestions for efficient training. Shilong Liu and Bojia Zi provided some significant technical suggestions on LLM tuning. Jiale Liu, Wenhao Yang, and Chenlai Qian provided some significant suggestions for us to polish the paper. Hongyang Li helped us a lot with the figure design. Yiren Pang provided GPT API keys when our keys were temporarily out of quota. The code is on the basis of Video-LLaVA, HumanTOMATO, MotionGPT. lit-gpt, and HumanML3D. Thanks to all contributors!

πŸ“š License

This code is distributed under an IDEA LICENSE. Note that our code depends on other libraries and datasets which each have their own respective licenses that must also be followed.

If you have any question, please contact at: thu [DOT] lhchen [AT] gmail [DOT] com AND shunlinlu0803 [AT] gmail [DOT] com.

🌟 Star History

Star History Chart

## πŸ“œ Citation ```bash @article{chen2024motionllm, title={MotionLLM: Understanding Human Behaviors from Human Motions and Videos}, author={Chen, Ling-Hao and Lu, Shunlin and Zeng, Ailing and Zhang, Hao and Wang, Benyou and Zhang, Ruimao and Zhang, Lei}, journal={arXiv preprint arXiv:2405.20340}, year={2024} } ```