add RT-DETR

yjh0410 · yjh0410 · commit 8ffa7be61cca · 2024-02-16T17:13:15.000+08:00
diff --git a/config/plain_detr_config.py b/config/plain_detr_config.py
@@ -76,8 +76,8 @@
         'lr_epoch': [11],     # 1x
         # ----------------- Input -----------------
         ## Transforms
-        'train_min_size': [320],   # short edge of image
-        'train_min_size2': [200,],
+        'train_min_size': [800],   # short edge of image
+        'train_min_size2': [400, 500, 600],
         'train_max_size': 1333,
         'test_min_size': 800,
         'test_max_size': 1333,
