前言
PyTorch3D 是 Facebook AI Research (FAIR) 开发的一个用于 3D 计算机视觉任务的开源深度学习库。它是建立在 PyTorch 框架之上的,并旨在为研究人员和工程师提供处理 3D 数据、渲染和几何计算的工具和模块。
PyTorch3D的用途:
-
3D 几何操作:PyTorch3D 提供了一系列用于处理 3D 几何数据的工具和函数。它允许用户进行点云操作、模型旋转、平移、缩放、重采样等。
-
3D 网格操作:该库支持处理 3D 网格数据,用户可以执行网格变换、网格采样、法线估计、网格融合等操作。
-
3D 渲染:PyTorch3D 提供了 3D 渲染引擎,允许用户在 3D 场景中渲染图像、视角的变换、光照和阴影等操作。
-
3D 数据加载和处理:用户可以使用 PyTorch3D 加载和处理不同格式的 3D 数据,包括 3D 模型、点云、深度图等。
-
PyTorch 集成:由于 PyTorch3D 是基于 PyTorch 开发的,因此用户可以利用 PyTorch 的自动微分机制进行梯度计算,便于深度学习任务的建模和训练。
PyTorch3D 的目标是为研究人员和开发者提供强大的 3D 计算机视觉工具,使其能够更容易地进行 3D 图像处理、渲染、几何计算以及深度学习模型的设计和训练。它适用于各种应用领域,包括计算机图形学、计算机视觉、机器学习和人工智能。你可以在 GitHub 上找到 PyTorch3D 的开源代码和文档,以深入了解其功能和用法。
安装Pytorch3D
安装pytorch3d,官方的要求是要有C++编译器:
gcc & g++ ≥ 4.9
那在在windows中,可以安装Visual Studio 来进行编译,我这里使用的是Visual Studio 2019 企业版本,安装时注意勾选下面的选项,安装完成之后重启电脑,如果在编译过程中提示找不Visual Studio,看下环境变量是否加上了Visual Studio。
1.官网安装方式
conda create -n pytorch3d python=3.9
conda activate pytorch3d
conda install pytorch=1.13.0 torchvision pytorch-cuda=11.6 -c pytorch -c nvidia
conda install -c fvcore -c iopath -c conda-forge fvcore iopath
pip install "git+https://github.com/facebookresearch/pytorch3d.git"
出现以下提示代表安装成功:
2.源码安装
下载pytorch3d源码,如果下载不了,按上面的百度网盘下载:https://pan.baidu.com/s/178WbPBRtCl3hF8b3niCeSA 提取码:icd8
git clone https://github.com/facebookresearch/pytorch3d.git
cd pytorch3d
python setup.py install
在安装pytorch3d可能出现错误,可以看文章结尾列的常见错误。
3.常见错误
subprocess.CalledProcessError: Command ‘[‘ninja’, ‘-v’]’ returned non-zero exit status 1.
出现这个错误时,找到dist-packages/torch/utils/cpp_extension.py这个文件,找到command = [‘ninja’, ‘-v’],改成 command = [‘ninja’, ‘–version’],改了效果如下:
# command = ['ninja', '-v']
command = ['ninja', '--version']
AttributeError: ‘Upsample’ object has no attribute ‘recompute_scale_factor’
把cuda改成11.7就可以解决这个错误。