From 97ac36ec6ceaab4a9a90934d33de3ddb8398e2f7 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 8 Nov 2019 10:19:46 -0800 Subject: [PATCH] updates --- train.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/train.py b/train.py index 263f3df4..34658fd3 100644 --- a/train.py +++ b/train.py @@ -60,6 +60,9 @@ def train(): hyp['cls_pw'] = 1. hyp['obj_pw'] = 1. + if img_size > 320: # scale hyp['obj'] by img_size (evolved at 320) + hyp['obj'] *= img_size / 320. + # Initialize init_seeds() multi_scale = opt.multi_scale @@ -407,7 +410,7 @@ if __name__ == '__main__': parser.add_argument('--bucket', type=str, default='', help='gsutil bucket') parser.add_argument('--img-weights', action='store_true', help='select training images by weight') parser.add_argument('--cache-images', action='store_true', help='cache images for faster training') - parser.add_argument('--weights', type=str, default='weights/yolov3-spp.weights', help='initial weights') # i.e. weights/darknet.53.conv.74 + parser.add_argument('--weights', type=str, default='weights/yolov3-spp.weights', help='initial weights') parser.add_argument('--arc', type=str, default='default', help='yolo architecture') # defaultpw, uCE, uBCE parser.add_argument('--prebias', action='store_true', help='transfer-learn yolo biases prior to training') parser.add_argument('--name', default='', help='renames results.txt to results_name.txt if supplied')