From fcea7fdd0e22362f1c0153a305ce9372d4dc3658 Mon Sep 17 00:00:00 2001 From: Wenyu Date: Wed, 23 Aug 2023 18:07:49 +0800 Subject: [PATCH] add rtdetr obj365, test=document_fix (#8567) --- configs/rtdetr/README.md | 3 +++ 1 file changed, 3 insertions(+) diff --git a/configs/rtdetr/README.md b/configs/rtdetr/README.md index 1b538110cc8..84237676bea 100644 --- a/configs/rtdetr/README.md +++ b/configs/rtdetr/README.md @@ -8,6 +8,7 @@ - 发布RT-DETR-R34模型 - 发布RT-DETR-R18模型 - 发布RT-DETR-Swin和RT-DETR-FocalNet模型 +- 发布RTDETR Obj365预训练模型 ## 简介 @@ -41,6 +42,8 @@ RT-DETR是第一个实时端到端目标检测器。具体而言,我们设计 ## Objects365预训练模型 | Model | Epoch | Dataset | Input shape | $AP^{val}$ | $AP^{val}_{50}$ | T4 TensorRT FP16(FPS) | Weight | Logs |:---:|:---:|:---:| :---:|:---:|:---:|:---:|:---:|:---:| +RT-DETR-R18 | 1x | Objects365 | 640 | 22.9 | 31.2 | - | [download](https://bj.bcebos.com/v1/paddledet/models/rtdetr_r18vd_1x_objects365.pdparams) | [log](https://github.com/lyuwenyu/RT-DETR/issues/8) +RT-DETR-R18 | 5x | COCO + Objects365 | 640 | 49.2 | 66.6 | 217 | [download](https://bj.bcebos.com/v1/paddledet/models/rtdetr_r18vd_5x_coco_objects365.pdparams) | [log](https://github.com/lyuwenyu/RT-DETR/issues/8) RT-DETR-R50 | 1x | Objects365 | 640 | 35.1 | 46.2 | - | [download](https://bj.bcebos.com/v1/paddledet/models/rtdetr_r50vd_1x_objects365.pdparams) | [log](https://github.com/lyuwenyu/RT-DETR/issues/8) RT-DETR-R50 | 2x | COCO + Objects365 | 640 | 55.3 | 73.4 | 108 | [download](https://bj.bcebos.com/v1/paddledet/models/rtdetr_r50vd_2x_coco_objects365.pdparams) | [log](https://github.com/lyuwenyu/RT-DETR/issues/8) RT-DETR-R101 | 1x | Objects365 | 640 | 36.8 | 48.3 | - | [download](https://bj.bcebos.com/v1/paddledet/models/rtdetr_r101vd_1x_objects365.pdparams) | [log](https://github.com/lyuwenyu/RT-DETR/issues/8)