mmpose/projects/mmpose4aigc/README_CN.md

2.9 KiB

MMPose for AIGC (AI Generated Content)

简体中文 | English

本项目将支持使用 MMPose 来生成骨架图片,用于姿态引导的 AI 图像生成。

当前已支持:

欢迎分享更多姿态引导的 AIGC 项目给我们!

快速上手

生成 Openpose 风格的骨架图片

Step 1: 准备

运行以下命令准备项目:

# install mmpose mmdet
pip install openmim
git clone https://github.com/open-mmlab/mmpose.git
cd mmpose
mim install -e .
mim install "mmdet>=3.0.0rc6"

# download models
bash download_models.sh

Step 2: 生成骨架图片

运行以下命令生成骨架图片:

bash mmpose_openpose.sh ../../tests/data/coco/000000000785.jpg

输入图片与生成骨架图片如下:

生成 MMPose 风格的骨架图片

Step 1: 准备

环境要求:

  • GCC >= 7.5
  • cmake >= 3.14

运行以下命令安装项目:

bash install_posetracker_linux.sh

最终的文件结构如下:

|----mmdeploy-1.0.0-linux-x86_64-cxx11abi
|    |----README.md
|    |----rtmpose-ort
|    |    |----rtmdet-nano
|    |    |----rtmpose-m
|    |    |----000000147979.jpg
|    |    |----t2i-adapter_skeleton.txt

Step 2: 生成姿态骨架图片

运行以下命令生成姿态骨架图片:

# 生成骨架图片
bash mmpose_style_skeleton.sh \
    mmdeploy-1.0.0-linux-x86_64-cxx11abi/rtmpose-ort/000000147979.jpg

更多详细信息可以查看 RTMPose

输入图片与生成骨架图片如下:

使用 T2I-Adapter

T2I- Adapter 在线试玩请点击 这里

Huggingface Gradio

结果展示