- Support SQR-DETR (CVPR 2023), credits to Fangyi Chen
- Support EVA-01 and EVA-02 backbones, please check DINO-EVA for more benchmarking results.
- Support CO-MOT aims for End-to-End Multi-Object Tracking by Feng Yan.
- Release
DINO
with optimized hyper-parameters which achieves50.0 AP
under 1x settings. - Release pretrained DINO based on
InternImage
,ConvNeXt-1K pretrained
backbones. - Release
Deformable-DETR-R50
pretrained weights. - Release
DETA
and betterH-DETR
pretrained weights: achieving50.2 AP
and49.1 AP
respectively.
- Support new algorithms including
Anchor-DETR
andDETA
. - Release more than 10+ pretrained models (including the converted weights):
DETR-R50 & R101
,DETR-R50 & R101-DC5
,DAB-DETR-R50 & R101-DC5
,DAB-DETR-R50-3patterns
,Conditional-DETR-R50 & R101-DC5
,DN-DETR-R50-DC5
,Anchor-DETR
and theDETA-Swin-o365-finetune
model which can achieve62.9AP
on coco val. - Support MaskDINO on ADE20k semantic segmentation task.
- Support
EMAHook
during training by settingtrain.model_ema.enabled=True
, which can enhance the model performance. DINO with EMA can achieve49.4AP
with only 12epoch training. - Support mixed precision training by setting
train.amp.enabled=True
, which will reduce 20% to 30% GPU memory usage. - Support
train.fast_dev_run=True
for fast debugging. - Support encoder-decoder checkpoint in DINO, which may reduce 30% GPU memory usage.
- Support a great slurm training scripts by @rayleizhu, please check this issue for more details #213
- MaskDINO COCO instance-seg/panoptic-seg pre-release #154
- New baselines for
Res/Swin-DINO-5scale
,ViTDet-DINO
,FocalNet-DINO
, etc. #138, #155 - Support FocalNet backbone #145
- Support Swin-V2 backbone #152
- Add ViTDet / FocalNet download links and usage example, please refer to Download Pretrained Weights.
- Add tutorial on how to verify the correct installation of detrex. #194
- Fix demo confidence filter not to remove mask predictions #156
- Make more readable logging info for criterion and matcher #151
- Modified learning rate scheduler config usage, add fundamental scheduler configuration #191
- Rebuild cleaner config files for projects #107
- Support H-Deformable-DETR #110
- Release H-Deformable-DETR pretrained weights including
H-Deformable-DETR-R50
,H-Deformable-DETR-Swin-Tiny
,H-Deformable-DETR-Swin-Large
. - Add demo for visualizing customized input images or videos using pretrained weights #119
- Release new baselines for
DINO-Swin-Large-36ep
,DAB-Deformable-DETR-R50-50ep
,DAB-Deformable-DETR-Two-Stage-50ep
,H-DETR
.
- Add model analyze tools for detrex #79
- Add benchmark #81
- Add visualization for COCO eval results and annotations #82
- Support
Group-DETR
algorhtim #84 - Release
DINO-Swin
training results #86 - Release better
Deformable-DETR
baselines #102 #103
- Fix bugs in ConvNeXt backbone #91
- Add pretrained model weights download links #86
The beta v0.1.0 version of detrex was released in 30/09/2022
- Support various backbones including: FocalNet, Swin-T, ResNet and other detectron2 builtin backbones.
- Add timm backbones wrapper and torchvision backbones wrapper.
- Support various transformer based detection algorithms including: DETR, Deformable-DETR, Conditional-DETR, DAB-DETR, DN-DETR, DINO.
- Support flexible config system based on Lazy Configs