Reinforcement learning from pixels is often bottlenecked by the performance and complexity of 3D rendered environments. Researchers face a trade-off between high-speed, low-level engines and slower, more accessible Python frameworks. To address this, we introduce PyBatchRender, a Python library for high-throughput, batched 3D rendering that achieves over 1 million FPS on simple scenes. Built on the Panda3D game engine, it utilizes its mature ecosystem while enhancing performance through optimized batched rendering for up to 1000X speedups. Designed as a physics-agnostic renderer for reinforcement learning from pixels, PyBatchRender offers greater flexibility than dedicated libraries, simpler setup than typical game-engine wrappers, and speeds rivaling state-of-the-art C++ engines like Madrona. Users can create custom scenes entirely in Python with tens of lines of code, enabling rapid prototyping for scalable AI training. Open-source and easy to integrate, it serves to democratize high-performance 3D simulation for researchers and developers. The library is available at https://github.com/dolphin-in-a-coma/PyBatchRender.


翻译:基于像素的强化学习常受限于三维渲染环境的性能与复杂性。研究人员不得不在高速但底层的引擎与较慢但更易用的Python框架之间进行权衡。为解决此问题,我们推出了PyBatchRender——一个用于高吞吐量批量三维渲染的Python库,在简单场景下可实现超过每秒一百万帧的渲染速度。该库基于Panda3D游戏引擎构建,在利用其成熟生态系统的同时,通过优化的批量渲染技术实现高达1000倍的性能提升。PyBatchRender专为基于像素的强化学习设计,作为物理无关的渲染器,它比专用库更具灵活性,比典型的游戏引擎封装更易配置,其速度可与Madrona等先进C++引擎相媲美。用户完全使用Python即可通过数十行代码创建自定义场景,从而为可扩展的AI训练实现快速原型开发。本库开源且易于集成,旨在为研究人员和开发者普及高性能三维仿真技术。项目地址:https://github.com/dolphin-in-a-coma/PyBatchRender。

0
下载
关闭预览

相关内容

【CVPR2024】VidLA: 大规模视频-语言对齐
专知会员服务
20+阅读 · 2024年3月31日
Python图像处理,366页pdf,Image Operators Image Processing in Python
图机器学习 2.2-2.4 Properties of Networks, Random Graph
图与推荐
10+阅读 · 2020年3月28日
CosFace: Large Margin Cosine Loss for Deep Face Recognition论文笔记
统计学习与视觉计算组
44+阅读 · 2018年4月25日
国家自然科学基金
3+阅读 · 2015年12月31日
国家自然科学基金
4+阅读 · 2015年12月31日
国家自然科学基金
46+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
2+阅读 · 2014年12月31日
VIP会员
相关基金
国家自然科学基金
3+阅读 · 2015年12月31日
国家自然科学基金
4+阅读 · 2015年12月31日
国家自然科学基金
46+阅读 · 2015年12月31日
国家自然科学基金
1+阅读 · 2015年12月31日
国家自然科学基金
2+阅读 · 2014年12月31日
Top
微信扫码咨询专知VIP会员