Skip to content
Permalink
Browse files

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

…to 0.7404, bird to 0.6951
  • Loading branch information...
RobinDong committed Jun 6, 2019
1 parent d662a9e commit 8dd17cd14706b221ae0e1cf2a5587f9db59f229c
Showing with 2 additions and 2 deletions.
  1. +1 −1 train.py
  2. +1 −1 utils/augmentations.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()
@@ -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)

0 comments on commit 8dd17cd

Please sign in to comment.
You can’t perform that action at this time.