modify train code

config/model_config/rtdetr_config.py

@@ -10,7 +10,7 @@ rtdetr_cfg = {
         'depth': 1.0,
         ## Image Encoder - Backbone
         'backbone': 'resnet18',
-        'backbone_norm': 'FrozeBN',
+        'backbone_norm': 'BN',
         'res5_dilation': False,
         'pretrained': True,
         'pretrained_weight': 'imagenet1k_v1',

dataset/data_augment/rtdetr_augment.py

@@ -75,7 +75,7 @@ class RandomPhotometricDistort(object):
         Returns:
             ndarray: the distorted image(s).
         """
-        if random.random() < 0.5:
+        if random.random() < 0.8:
             dhue = np.random.uniform(low=-self.hue, high=self.hue)
             dsat = self._rand_scale(self.saturation)
             dexp = self._rand_scale(self.exposure)

models/detectors/rtdetr/build.py

@@ -20,7 +20,7 @@ def build_rtdetr(args, cfg, num_classes=80, trainable=False, deploy=False):
     model = RT_DETR(cfg             = cfg,
                     num_classes     = num_classes,
                     conf_thresh     = args.conf_thresh,
-                    topk            = 100,
+                    topk            = 300,
                     deploy          = deploy,
                     no_multi_labels = args.no_multi_labels,
                     )