SimpleDet:一个简洁通用的分布式目标检测与实例识别框架
计算机视觉与模式识别
2019-03-15 v1 机器学习
摘要
目标检测和实例识别在自动驾驶、视频监控和医学图像分析等许多 AI 应用中起着核心作用。然而,在大规模数据集上训练目标检测模型仍然计算昂贵且耗时。本文提出一个高效且开源的目标检测框架 SimpleDet,它能够在消费级硬件上大规模训练最先进的检测模型。SimpleDet 支持具备最佳实践的近期检测模型。SimpleDet 还支持近乎线性扩展的分布式训练,开箱即用。SimpleDet 的代码、示例和文档可在 https://github.com/tusimple/simpledet 获取。
引用
@article{arxiv.1903.05831,
title = {SimpleDet: A Simple and Versatile Distributed Framework for Object Detection and Instance Recognition},
author = {Yuntao Chen and Chenxia Han and Yanghao Li and Zehao Huang and Yi Jiang and Naiyan Wang and Zhaoxiang Zhang},
journal= {arXiv preprint arXiv:1903.05831},
year = {2019}
}
备注
Tech Report