Blendtorch深度学习 Python 框架

联合创作 · 2023-09-26

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

功能总结

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

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

安装

Blendtorch 是由两个不同的子包:bendtorch.bttblendtorch.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 

确保 Blender 可执行文件在你的环境中查找PATH。在 Windows 上,这可以通过

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

完整的 Blender 设置

至少打开一次 Blender,并完成初始设置。如果错过这一步,某些测试(尤其是与 RL 相关的测试)将失败(Blender 2.91)。

安装 Blendtorch Blender 部分

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

安装 Blendtorch PyTorch 部分

pip install -e /pkg_pytorch 

安装blendtorch-btt到您打算从中运行 PyTorch 的 Python 环境中。虽然不是必需的,但如果您打算使用 Blendtorch 进行强化学习,建议安装 OpenAI Gym

pip install gym 

 

浏览 4
点赞
评论
收藏
分享

手机扫一扫分享

编辑
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

编辑
举报