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
- [2024-06-17]: MoVid dataset Video data is now available at HuggingFace.
- [2024-06-11]: CLI mode is supported. MotionLLM is running on A100 GPU on π€HuggingFace demo (Posts on Twitter).
- [2024-05-31]: Paper, demo, and codes are released (Posts on Twitter).
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.
We provide a simple online demo for you to try MotionLLM. Below is the guidance to deploy the demo on your local machine.
pip install -r requirements.txt
2.1 Download the LLM
Please follow the instruction of Lit-GPT to prepare the LLM model (vicuna 1.5-7B). These files will be:
./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). Opening for the suggestions to Ling-Hao Chen and Shunlin Lu.
wget xxx
Keep them in a folder named and remember the path (LINEAR_V
and LORA
).
Choice 1: gradio demo
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.
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.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.
# Example here
Input video path: xxx.mp4
Your question: what xxx ?
================================
The man plan to xxx.
================================
- Release the CLI demo of MotionLLM.
- Release the video demo of MotionLLM.
- Release the motion demo of MotionLLM.
- Release the MoVid dataset and MoVid-Bench.
- Release the tuning instruction of MotionLLM.
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!
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.
@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}
}