当前位置：首页 > news >正文

ExPose推理脚本全解析：如何批量处理图像并导出3D模型参数与网格数据

news 2026/8/15 15:44:32

ExPose推理脚本全解析：如何批量处理图像并导出3D模型参数与网格数据

【免费下载链接】exposeExPose - EXpressive POse and Shape rEgression项目地址: https://gitcode.com/gh_mirrors/expo/expose

ExPose（EXpressive POse and Shape rEgression）是一款强大的3D人体姿态与形状回归工具，能够从单张图像中精准提取人体3D模型参数并生成网格数据。本文将详细介绍如何使用ExPose的推理脚本实现图像批量处理，帮助新手快速掌握从环境配置到结果导出的完整流程。

📋 准备工作：环境配置与依赖安装

在开始使用ExPose推理脚本前，需确保系统满足以下环境要求并安装必要依赖：

系统要求

操作系统：Linux
硬件：NVIDIA GPU（支持CUDA）
Python版本：3.6+

快速安装步骤

克隆仓库

git clone https://gitcode.com/gh_mirrors/expo/expose cd expose

安装依赖ExPose的核心依赖已在requirements.txt中列出，包括PyTorch、Open3D、SMPLX等：
```
pip install -r requirements.txt
```
关键依赖说明：
- torch>=1.6.0：深度学习框架
- open3d>=0.10.0.0：3D模型可视化与网格处理
- smplx>=0.1.21：人体参数化模型
- pyrender>=0.1.43：3D渲染引擎

🚀 核心脚本解析：inference.py与demo.py

ExPose提供了两个主要推理脚本，分别适用于不同场景：

1. inference.py：批量处理数据集

inference.py主要用于处理标准数据集（如3DPW、MPI-INF-3DHP等），支持批量推理与结果导出。其核心功能包括：

加载预训练模型与配置文件
读取数据集并进行预处理
执行3D姿态与形状回归
导出参数文件（.npz）与网格数据（.ply）

关键参数说明：

--exp-cfg # 实验配置文件路径 --output-folder # 结果保存目录 --save-params # 是否保存3D模型参数 --save-mesh # 是否保存网格数据 --datasets # 指定处理的数据集

2. demo.py：自定义图像文件夹处理

demo.py适用于处理用户自定义图像文件夹，内置R-CNN目标检测模块自动检测图像中的人体区域。相比inference.py，它增加了以下功能：

自动人体检测与边界框提取
支持单张或批量图像输入
可视化结果实时显示

关键参数差异：

--image-folder # 自定义图像文件夹路径 --rcnn-batch # R-CNN检测批次大小 --show # 是否实时显示结果

🔍 推理流程详解

ExPose的推理过程可分为四个核心步骤，以下以demo.py为例进行说明：

步骤1：图像预处理与人体检测

# demo.py 第75-157行 def preprocess_images(image_folder, exp_cfg): # 加载R-CNN模型进行人体检测 rcnn_model = keypointrcnn_resnet50_fpn(pretrained=True) # 检测人体边界框并裁剪 expose_dset = ImageFolderWithBoxes(img_paths, bboxes, transforms=transforms) return dutils.DataLoader(expose_dset, batch_size=batch_size)

此步骤通过预训练的Keypoint R-CNN模型检测图像中的人体区域，生成边界框并进行裁剪，为后续3D推理做准备。

步骤2：模型加载与推理

# demo.py 第244-262行 model = SMPLXNet(exp_cfg).to(device) checkpointer = Checkpointer(model, save_dir=checkpoint_folder) checkpointer.load_checkpoint() # 加载预训练权重 model_output = model(body_imgs, body_targets, full_imgs=full_imgs)

ExPose模型SMPLXNet由多个子网络组成，包括：

主干网络（ResNet/HRNet）：提取图像特征
姿态回归网络：预测人体关节角度
形状回归网络：预测人体形状参数（betas）
相机参数回归：预测相机内参和外参

步骤3：3D结果可视化

ExPose从单张图像生成的3D人体姿态与形状可视化结果

可视化通过HDRenderer实现，支持：

原始图像与3D网格叠加显示
多视角旋转渲染（通过--degrees参数控制）
手部、头部区域单独可视化

步骤4：结果导出

ExPose支持三种结果格式导出，通过命令行参数控制：

可视化图像（--save-vis）保存叠加3D网格的图像，路径格式：{output_folder}/{image_name}/hd_overlay.png

3D网格数据（--save-mesh）以PLY格式保存3D网格，可用于Blender等3D建模软件：

# demo.py 第418-431行 expose_mesh = o3d.geometry.TriangleMesh() expose_mesh.vertices = Vec3d(final_model_vertices[idx] + hd_params['transl'][idx]) o3d.io.write_triangle_mesh(mesh_fname, expose_mesh)

模型参数（--save-params）保存SMPLX模型参数（姿态、形状、相机等）到NPZ文件，包含：
- vertices：3D顶点坐标
- betas：形状参数
- body_pose：身体姿态参数
- camera_translation：相机平移

💻 实战案例：批量处理图像并导出3D模型

以下是使用demo.py处理自定义图像文件夹并导出网格数据的完整命令：

python demo.py \ --image-folder ./samples \ --exp-cfg ./expose/config/default.yaml \ --output-folder ./results \ --save-vis true \ --save-mesh true \ --save-params true \ --degrees 0 90 180 270 \ --expose-batch 2