Enhance expand scale to 6 and reduce overlap_thres to 0.4. Boost mAP …

…to 0.7404, bird to 0.6951
RobinDong · Jun 6, 2019 · 8dd17cd · 8dd17cd
1 parent d662a9e
commit 8dd17cd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/train.py b/train.py
@@ -126,7 +126,7 @@ def train():
 
     optimizer = optim.SGD(net.parameters(), lr=args.lr, momentum=args.momentum,
                           weight_decay=args.weight_decay)
-    criterion = MultiBoxLoss(cfg['num_classes'], 0.5, True, 0, True, 3, 0.5,
+    criterion = MultiBoxLoss(cfg['num_classes'], 0.4, True, 0, True, 3, 0.5,
                              False, args.cuda)
 
     net.train()

diff --git a/utils/augmentations.py b/utils/augmentations.py
@@ -318,7 +318,7 @@ def __call__(self, image, boxes, labels):
             return image, boxes, labels
 
         height, width, depth = image.shape
-        ratio = random.uniform(1, 4)
+        ratio = random.uniform(1, 6)
         left = random.uniform(0, width*ratio - width)
         top = random.uniform(0, height*ratio - height)