Blendtorch深度學(xué)習(xí) Python 框架
Blendtorch是一個(gè) Python 框架，可將 Blender 無(wú)縫集成到 PyTorch 數(shù)據(jù)集，以便從人工視覺數(shù)據(jù)中進(jìn)行深度學(xué)習(xí)。利用 Eevee，一種新的基于物理的實(shí)時(shí)渲染器，實(shí)時(shí)合成圖像和注釋，從而避免在許多情況下停止模型訓(xùn)練。
功能總結(jié)
- 數(shù)據(jù)流：實(shí)時(shí)將分布式 Blender 渲染直接流式傳輸?shù)?PyTorch 數(shù)據(jù)管道中。
- OpenAI Gym 支持：創(chuàng)建和運(yùn)行遠(yuǎn)程控制的 Blender Gym 來(lái)訓(xùn)練強(qiáng)化代理。
下圖顯示了在為現(xiàn)實(shí)世界檢測(cè)任務(wù)生成人工訓(xùn)練數(shù)據(jù)的上下文中使用的Blendtorch的基本概念。
安裝
Blendtorch 是由兩個(gè)不同的子包：bendtorch.btt和blendtorch.btb組成。
先決條件
- Blender >= 2.83/2.91 (Python 3.7)
- PyTorch >= 1.50 (Python 3.7/3.8) 運(yùn)行 Windows 10 和 Linux
克隆這個(gè)倉(cāng)庫(kù)
git clone https://github.com/cheind/pytorch-blender.git
確保 Blender 可執(zhí)行文件在你的環(huán)境中查找PATH。在 Windows 上，這可以通過(guò)
set PATH=c:\\Program Files\\Blender Foundation\\Blender 2.91;%PATH%
完整的 Blender 設(shè)置
至少打開一次 Blender，并完成初始設(shè)置。如果錯(cuò)過(guò)這一步，某些測(cè)試（尤其是與 RL 相關(guān)的測(cè)試）將失?。˙lender 2.91）。
安裝 Blendtorch Blender 部分
blender --background --python /scripts/install_btb.py
安裝 Blendtorch PyTorch 部分
pip install -e /pkg_pytorch
安裝blendtorch-btt到您打算從中運(yùn)行 PyTorch 的 Python 環(huán)境中。雖然不是必需的，但如果您打算使用 Blendtorch 進(jìn)行強(qiáng)化學(xué)習(xí)，建議安裝 OpenAI Gym
pip install gym
評(píng)論
圖片
表情