项目地址:https://github.com/YuliangXiu/ICON
项目描述:
照片生成穿衣服的3d人体模型
安装步骤:
克隆项目:
git clone https://github.com/YuliangXiu/ICON
创建新环境:
conda create -n vhuman_icon39 python=3.9.0
使用3.9版本的python,不能使用3.12的,因为后面有的库不支持3.12
启动环境:
activate vhuman_icon39
安装pytorch:
在官网选择跟自己电脑配置相同的pytorch 官网:https://pytorch.org/get-started/locally/
安装最新版pytorch2.2.1即可,我的配置是cuda12.1
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
安装其它依赖:
打开克隆的icon项目,缺少哪个库就用pip install安装哪个即可(因为需要运行apps文件夹中的infer.py,所以直接在编译器里打开这个文件,运行这个文件,报哪个库缺少的错,就安装哪个库就行)
下面是几个不能用pip install命令直接按名字安装的
pytorch3d:
pip install "git+https://github.com/facebookresearch/pytorch3d.git"
后面pytorch3d还会报错:No module named ‘pywintypes‘
这是因为pywin32有问题,直接重装pywin32
pip uninstall pywin32
pip install pywin32
cv2:
pip install opencv-python
_pickle:
不需要安装,只是在3.11里换了个名字,把_pickle改成pickle即可
# import pickle as cPickle
import _pickle as cPickle
kaolin:
这个不能直接用pip安装,安装出来的那个是错的。提示去官网安装 官网:https://kaolin.readthedocs.io/en/latest/notes/installation.html
官网提供的安装方式如下:
第一个方法,直接pip安装:(我没安装成功)
pip install kaolin==0.15.0 -f https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-{TORCH_VER}_cu{CUDA_VER}.html
其中的{TORCH_VER}换成pytorch版本,{CUDA_VER}换成cuda版本。按照自己的来。比如我安装的是pytorch2.2.1 cuda12.1,就按下面这么写
pip install kaolin==0.15.0 -f https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.2.1_cu121.html
但是报错了,说0.15.0版本不支持,我把0.15.0删掉了,不指定版本安装,然后装上了0.1.0版本,然后发现这个版本不行
第二个方法,使用源码安装:
首先找个文件夹克隆kaolin kaolin源码:https://github.com/NVIDIAGameWorks/kaolin
git clone --recursive https://github.com/NVIDIAGameWorks/kaolin
用编译器打开克隆的kaolin文件夹(记得打开刚才conda创建的vhuman_icon39环境)
安装kaolin所需的依赖:
pip install -r tools/build_requirements.txt -r tools/viz_requirements.txt -r tools/requirements.txt
安装kaolin
python setup.py develop
tinyobjloader:
需要用源码安装 源码:https://github.com/tinyobjloader/tinyobjloader/blob/release/python/README.md
随便找个地方clone
git clone https://github.com/tinyobjloader/tinyobjloader
进入文件夹(不建议在git bush里这样操作,因为无法直接使用conda)
建议直接用编译器打开tinyobjloader文件夹
记得使用刚才conda创建的vhuman_icon39环境,不然装不进咱们需要的环境里(activate vhuman_icon39 进行使用,或者vscode右下角直接选环境)
python -m pip install .
(最后那个点别忘了复制)
sklearn:
pip install scikit-learn
其它的包都可以直接pip install,就不一一赘述,遇到报错缺少xx模块,直接pip install xx就行了
下一步:
还是不能运行,因为还缺模型,该安装需要的模型了
下一篇:
https://blog.csdn.net/weixin_45615730/article/details/136971865?spm=1001.2014.3001.5501