首页 > 快讯 > 快讯详情

智元开源高性能视频数据加载库Video Datase

格隆汇12月9日|智元机器人宣布开源Video Dataset。Video Dataset是智元机器人团队基于实际AI训练需求深度开发的高性能视频数据加载库。通过将解码任务从传统的CPU转移到GPU,充分挖掘硬件解码能力,吞吐量提升4倍。同时兼容PyTorch Dataset接口,提供Mixin类,开发者改几行代码即可接入现有训练流。