【AI】Windows环境安装SPAR3D单图三维重建心得

效果一览

左图为原始单个图像，右图为通过SPAR3D重建后的三维建模，可以看出效果还是不错的。

本地环境配置

• 系统：Windows 11 专业版
• CPU：i5-13400F
• 内存：32GB
• GPU：RTX3060 12GB
• cuda：11.8
• conda：23.7.4

安装步骤

项目地址（GitHub - Stability-AI/stable-point-aware-3d: SPAR3D: Stable Point-Aware Reconstruction of 3D Objects from Single Images）

前置依赖

• Python >= 3.8
• 对于Windows环境，若使用GPU则需要CUDA
• 对于Windows环境，由于需要使用c生成依赖，必须提前安装Visual Studio 2022及对应MSVC工具
• 提前根据cuda版本在conda环境安装对应Pytorch（一定要对应）

Windows安装步骤

• 下载代码，conda创建环境并进入（这里使用python3.10）

conda create -n spar3d python=3.10

• 安装项目依赖，若遇到git访问超时问题，建议注释requirements对应git，单独对注释依赖进行git clone下载手动进入文件夹使用pip install .

pip install -r requirements.txt

• 由于本人环境实际推理有报错问题，建议直接同时安装remesh的依赖

pip install -r requirements-remesh.txt

• 根据要求补充安装依赖

pip install -U setuptools==69.5.1
pip install wheel

运行

• 因需要提前下载一些模型，若有huggingface.co访问问题，在run.py头两行加入镜像配置

import os
os.environ["HF_ENDPOINT"] = "https://hf-mirror.com"

• 因下载对应预训练模型需要注册，请自行科学上网并搬运模型（约7GB），本文存在在项目/models/spar3d文件夹中
• 执行推理命令，查看建模效果

# pretrained-model必须指定为当前项目相对路径文件夹，或自行修改代码实现绝对路径
# 建议使用low-vram-mode减少显存占用，测试发现12G显存不使用容易执行不成功
python run.py demo_files/examples/fish.png --pretrained-model models/spar3d  --low-vram-mode --output-dir output/