detrex v0.1.1 Release

rentainhe released this 18 Oct 11:04

· 117 commits to main since this release

What's New

Add model analysis tools in tools.
Support visualization on COCO eval results and annotations in tools.
Support Group-DETR.
Release more DINO training results in DINO.
Release better Deformable-DETR baselines in Deformable-DETR.
Fix ConvNeXt bugs.
Add pretrained weights download links and usage in documentation, see Download Pretrained Backbone Weights.
Add documentation for tools, see Practical Tools and Scripts.

New Pretrained Models

All the pretrained weights can be downloaded in detrex Model Zoo.

DINO

Method	Pretrained	Epochs	Box AP
DINO-R50-4Scale	IN1k	24	50.60
DINO-R101-4Scale	IN1k	12	49.95
DINO-Swin-Tiny-224-4Scale	IN1k	12	51.30
DINO-Swin-Tiny-224-4Scale	IN22k to IN1k	12	51.30
DINO-Swin-Small-224-4Scale	IN1k	12	52.96
DINO-Swin-Base-384-4Scale	IN22k to IN1k	12	55.83
DINO-Swin-Large-224-4Scale	IN22k to IN1k	12	56.92
DINO-Swin-Large-384-4Scale	IN22k to IN1k	12	56.93

Deformable-DETR

Method	Pretrained	Epochs	Box AP
Deformable-DETR-R50 + Box-Refinement	IN1k	50	46.99
Deformable-DETR-R50 + Box-Refinement + Two-Stage	IN1k	50	48.19

Assets 2