Change scales formula for Default Boxes.

Author: datumbox

test/expect/ModelTester.test_ssdlite320_mobilenet_v3_large_expect.pkl

@@ -138,17 +138,19 @@ class DefaultBoxGenerator(nn.Module):
     Args:
         aspect_ratios (List[List[int]]): A list with all the aspect ratios used in each feature map.
         min_ratio (float): The minimum scale :math:`\text{s}_{\text{min}}` of the default boxes used in the estimation
-            of the scales of each feature map.
+            of the scales of each feature map. It is used only if the ``scales`` parameter is not provided.
         max_ratio (float): The maximum scale :math:`\text{s}_{\text{max}}`  of the default boxes used in the estimation
-            of the scales of each feature map.
+            of the scales of each feature map. It is used only if the ``scales`` parameter is not provided.
+        scales (List[float]], optional): The scales of the default boxes. If not provided it will be estimated using
+            the ``min_ratio`` and ``max_ratio`` parameters.
         steps (List[int]], optional): It's a hyper-parameter that affects the tiling of defalt boxes. If not provided
             it will be estimated from the data.
         clip (bool): Whether the standardized values of default boxes should be clipped between 0 and 1. The clipping
             is applied while the boxes are encoded in format ``(cx, cy, w, h)``.
     """
 
     def __init__(self, aspect_ratios: List[List[int]], min_ratio: float = 0.15, max_ratio: float = 0.9,
-                 steps: Optional[List[int]] = None, clip: bool = True):
+                 scales: Optional[List[float]] = None, steps: Optional[List[int]] = None, clip: bool = True):
         super().__init__()
         if steps is not None:
             assert len(aspect_ratios) == len(steps)
@@ -158,15 +160,12 @@ def __init__(self, aspect_ratios: List[List[int]], min_ratio: float = 0.15, max_
         num_outputs = len(aspect_ratios)
 
         # Estimation of default boxes scales
-        # Inspired from https://github.com/weiliu89/caffe/blob/ssd/examples/ssd/ssd_pascal.py#L311-L317
-        min_centile = int(100 * min_ratio)
-        max_centile = int(100 * max_ratio)
-        conv4_centile = min_centile // 2  # assume half of min_ratio as in paper
-        step = (max_centile - min_centile) // (num_outputs - 2)
-        centiles = [conv4_centile, min_centile]
-        for c in range(min_centile, max_centile + 1, step):
-            centiles.append(c + step)
-        self.scales = [c / 100 for c in centiles]
+        if scales is None:
+            range_ratio = max_ratio - min_ratio
+            self.scales = [min_ratio + range_ratio * k / (num_outputs - 1.0) for k in range(num_outputs)]
+            self.scales.append(1.0)
+        else:
+            self.scales = scales
 
         self._wh_pairs = []
         for k in range(num_outputs):

torchvision/models/detection/anchor_utils.py

@@ -552,7 +552,9 @@ def ssd300_vgg16(pretrained: bool = False, progress: bool = True, num_classes: i
         pretrained_backbone = False
 
     backbone = _vgg_extractor("vgg16_features", False, progress, pretrained_backbone, trainable_backbone_layers, True)
-    anchor_generator = DefaultBoxGenerator([[2], [2, 3], [2, 3], [2, 3], [2], [2]], steps=[8, 16, 32, 64, 100, 300])
+    anchor_generator = DefaultBoxGenerator([[2], [2, 3], [2, 3], [2, 3], [2], [2]],
+                                           scales=[0.07, 0.15, 0.33, 0.51, 0.69, 0.87, 1.05],
+                                           steps=[8, 16, 32, 64, 100, 300])
     model = SSD(backbone, anchor_generator, (300, 300), num_classes,
                 image_mean=[0.48235, 0.45882, 0.40784], image_std=[1., 1., 1.], **kwargs)
     if pretrained: