detrex 项目介绍
detrex 是一个开源工具箱,专注于提供最先进的基于 Transformer 的检测算法。该项目建立在 Facebook 的 Detectron2 之上,部分模块设计借鉴自 MMDetection 和 DETR。detrex 旨在为用户提供高性能和易用的检测模型。
项目特色
- 模块化设计:detrex 将基于 Transformer 的检测框架分解为多个组件,方便用户轻松定制自己的模型。
- 强大的基线:detrex 提供一系列的基准模型,并通过优化大多数支持算法中的超参数,将模型性能从 0.2 AP 提升到 1.1 AP。
- 易于使用:detrex 设计轻量且易于使用,采用了 LazyConfig 系统来提供更灵活的语法和更简洁的配置文件,并且配备了轻量级的训练引擎。
除了 detrex 本身,我们还推出了一个名为 Awesome Detection Transformer 的仓库,用来展示关于 Transformer 在检测与分割中的使用的学术论文。
项目名称的趣味解释
- detr-ex: 此名寓意向 DETR 致敬,将本仓库视为基于 Transformer 检测算法的扩展。
- det-rex: 在拉丁语中,rex 意为 "国王"。我们希望这个仓库能推动车辆检测领域的进步。
- de-t.rex: 在荷兰语中,"de" 意为 "the",而 T.rex(即霸王龙)寓意 "暴君蜥蜴之王",与我们的研究工作 "DINO"(恐龙)相呼应。
最新动态
在 2023 年 7 月 16 日发布的 v0.5.0 版本中新增支持:
- Focus-DETR
- SQR-DETR
- Align-DETR
- EVA-01 和 EVA-02 骨干网络(请查看 DINO-EVA 了解更多基准测试结果)
入门指南
请参考 detrex 的入门指南以获取 detrex 的基本使用信息。我们还提供了其他的教学指导,包括配置系统、如何转换预训练权重、如何在 COCO 数据集上可视化训练数据和测试结果等。
模型库
detrex 的模型库包含了多种已经实现的方法,包括:
- DETR、Deformable-DETR、PnP-DETR 等克拉斯信息。
许可证
该项目遵循 Apache 2.0 许可证。
致谢
detrex 是由 IDEA 研究团队创建的开源工具箱,构建在 Detectron2 之上,并部分借鉴了 MMDetection、DETR 和 Deformable-DETR 模块的设计。我们对所有贡献者表示感谢!
以最简单的语言介绍 detrex 项目,detrex 提供一个开放的平台,供研究人员和开发者研究和应用最新的 Transformer 检测技术。通过其模块化设计,用户可以快速进行模型定制,适用于各类检测任务,致力于推动最前沿的目标检测算法在学术界和工业界的发展。