开源Transformer检测算法工具箱
detrex是一个开源工具箱,专为最先进的Transformer检测算法提供支持。该工具箱基于Detectron2构建,并参考了MMDetection和DETR的模块设计。detrex模块化设计,提供强大基线,通过优化超参数将模型性能提升至0.2至1.1AP。该工具箱轻量易用,支持最新算法如Focus-DETR、SQR-DETR、Align-DETR、EVA-01和EVA-02,帮助用户构建定制模型。
detrex 是一个开源工具箱,专注于提供最先进的基于 Transformer 的检测算法。该项目建立在 Facebook 的 Detectron2 之上,部分模块设计借鉴自 MMDetection 和 DETR。detrex 旨在为用户提供高性能和易用的检测模型。
除了 detrex 本身,我们还推出了一个名为 Awesome Detection Transformer 的仓库,用来展示关于 Transformer 在检测与分割中的使用的学术论文。
在 2023 年 7 月 16 日发布的 v0.5.0 版本中新增支持:
请参考 detrex 的入门指南以获取 detrex 的基本使用信息。我们还提供了其他的教学指导,包括配置系统、如何转换预训练权重、如何在 COCO 数据集上可视化训练数据和测试结果等。
detrex 的模型库包含了多种已经实现的方法,包括:
该项目遵循 Apache 2.0 许可证。
detrex 是由 IDEA 研究团队创建的开源工具箱,构建在 Detectron2 之上,并部分借鉴了 MMDetection、DETR 和 Deformable-DETR 模块的设计。我们对所有贡献者表示感谢!
以最简单的语言介绍 detrex 项目,detrex 提供一个开放的平台,供研究人员和开发者研究和应用最新的 Transformer 检测技术。通过其模块化设计,用户可以快速进行模型定制,适用于各类检测任务,致力于推动最前沿的目标检测算法在学术界和工业界的发展。