MusicInfuser:一个可以根据音乐和文本生成舞蹈视频的模型

MusicInfuser 是什么?

MusicInfuser 是一个将文本到视频扩散模型与音乐对齐的项目,能够根据音乐和文本生成舞蹈视频。它通过引入音乐-视频交叉注意力机制,使生成的视频能够与音乐节奏同步。MusicInfuser:一个可以根据音乐和文本生成舞蹈视频的模型.jpg

MusicInfuser 功能

  • 音乐驱动的舞蹈视频生成:根据输入的音乐和文本提示生成高质量的舞蹈视频。

  • 灵活的文本提示:用户可以通过文本提示指定舞蹈风格、场景、舞者特征等。

  • 支持长序列生成:虽然模型最初是基于 73 帧序列训练的,但可以外推生成更长的视频序列。

  • 随机种子控制:通过设置随机种子,用户可以生成不同的舞蹈结果。

MusicInfuser 应用

  • 创意视频制作:为视频创作者提供音乐驱动的舞蹈视频生成工具,提升内容创作效率。

  • 娱乐与表演:用于生成个性化的舞蹈表演视频,适用于虚拟现实、增强现实等场景。

  • 教育与培训:帮助舞蹈学习者通过生成的视频更好地理解舞蹈动作与音乐节奏的配合。

MusicInfuser 使用方法

一、安装

1. 克隆仓库:

git clone https://github.com/SusungHong/MusicInfuser
cd MusicInfuser

2. 创建并激活 Conda 环境:

conda create -n musicinfuser python=3.10
conda activate musicinfuser

3. 安装依赖:

pip install -r requirements.txt
pip install -e ./mochi --no-build-isolation

4. 下载模型权重:

python ./music_infuser/download_weights.py weights/

二、推理(生成视频)

运行以下命令生成视频:

python inference.py --input-file {MP3或MP4文件路径} \
                    --prompt {文本提示} \
                    --num-frames {生成的帧数}
  • input-file:输入文件(MP3 或 MP4),从中提取音频。

  • prompt:生成舞蹈的文本提示。提示越具体,生成结果通常越好,但过于具体可能会降低音频的影响。默认值为 "a professional female dancer dancing K-pop in an advanced dance setting in a studio with a white background, captured from a front view"。

  • num-frames:生成的帧数。默认值为 145。

其他可选参数:

  • seed:随机种子,用于控制生成结果的随机性。默认值为 None。

  • cfg-scale:文本提示的分类器自由引导(CFG)比例。默认值为 6.0。

三、训练

1. 数据预处理:

bash music_infuser/preprocess.bash -v {数据集路径} -o {处理后的视频输出目录} -w {预训练的 Mochi 模型路径} --num_frames {帧数}

2. 运行训练:

bash music_infuser/run.bash -c music_infuser/configs/music_infuser.yaml -n 1

注意:当前实现仅支持单 GPU 训练,训练 73 帧序列需要大约 80GB 的显存。

相关链接

GitHub 项目页面:https://github.com/SusungHong/MusicInfuser

论文:https://arxiv.org/abs/2503.14505

收藏
#文字转视频
最新工具
Awesome LangGraph
Awesome LangGraph

一份精心整理的 LangGraph 资源合集,涵盖了基础介绍、入...

VDSpeak
VDSpeak

一个专注于为YouTube视频提供多语言翻译和配音服务的在线工具...

PydanticAI
PydanticAI

一个Pydantic 团队开发的Python代理框架,可以通过生...

AutoGLM沉思
AutoGLM沉思

智谱推出的一款集深度研究与实际操作能力于一体的AI Agent产...

JustAI
JustAI

一款AI文案写作与营销工具,能生成多场景文案,包括小红书笔记、抖...

巧文书
巧文书

原通答AI,一款专为企业打造的AI智能写标书工具,通过AI大模型...

Ai2 Paper Finder
Ai2 Paper Finder

一款由艾伦AI研究所发布的基于LLM的文献搜索系统,Ai2 Pa...

Vivago AI
Vivago AI

北京智象未来科技有限公司面向全球市场推出的一款综合性在线 AI ...

Ghibli AI
Ghibli AI

一款专门用于生成吉卜力风格的AI图像生成器,能够根据用户的文本描...

快出题
快出题

金数据旗下的一款AI驱动的在线考试软件,能够快速生成题库并自动组...