Blendtorch深度学习 Python 框架

Blendtorch是一个 Python 框架，可将 Blender 无缝集成到 PyTorch 数据集，以便从人工视觉数据中进行深度学习。利用 Eevee，一种新的基于物理的实时渲染器，实时合成图像和注释，从而避免在许多情况下停止模型训练。

功能总结

• 数据流：实时将分布式 Blender 渲染直接流式传输到 PyTorch 数据管道中。
• OpenAI Gym 支持：创建和运行远程控制的 Blender Gym 来训练强化代理。

下图显示了在为现实世界检测任务生成人工训练数据的上下文中使用的Blendtorch的基本概念。

安装

Blendtorch 是由两个不同的子包：bendtorch.btt和blendtorch.btb组成。

先决条件

• Blender >= 2.83/2.91 (Python 3.7)
• PyTorch >= 1.50 (Python 3.7/3.8) 运行 Windows 10 和 Linux

克隆这个仓库

git clone https://github.com/cheind/pytorch-blender.git 

set PATH=c:\\Program Files\\Blender Foundation\\Blender 2.91;%PATH%

blender --background --python /scripts/install_btb.py 

pip install -e /pkg_pytorch 

pip install gym