modify rtdetr_r18 config

config/model_config/rtdetr_config.py

@@ -10,12 +10,12 @@ rtdetr_cfg = {
         'depth': 1.0,
         ## Image Encoder - Backbone
         'backbone': 'resnet18',
-        'backbone_norm': 'BN',
+        'backbone_norm': 'FrozeBN',
         'res5_dilation': False,
         'pretrained': True,
         'pretrained_weight': 'imagenet1k_v1',
-        'freeze_at': -1,
-        'freeze_stem_only': True,
+        'freeze_at': 0,
+        'freeze_stem_only': False,
         'out_stride': [8, 16, 32],
         'max_stride': 32,
         ## Image Encoder - FPN

dataset/data_augment/rtdetr_augment.py

@@ -75,7 +75,7 @@ class RandomPhotometricDistort(object):
         Returns:
             ndarray: the distorted image(s).
         """
-        if random.random() < 0.8:
+        if random.random() < 0.5:
             dhue = np.random.uniform(low=-self.hue, high=self.hue)
             dsat = self._rand_scale(self.saturation)
             dexp = self._rand_scale(self.exposure)
@@ -124,8 +124,6 @@ class RandomSampleCrop(object):
             (0.3, None),
             (0.7, None),
             (0.9, None),
-            # randomly sample a patch
-            (None, None),
         )
 
     def intersect(self, box_a, box_b):