From bd2d3cc5d174f6339ab58ce2302d456fa71196d9 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 6 Aug 2019 14:35:18 +0200 Subject: [PATCH] updates --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index f8f04e58..bc07e3e9 100644 --- a/train.py +++ b/train.py @@ -205,7 +205,7 @@ def train(cfg, model_info(model, report='summary') # 'full' or 'summary' nb = len(dataloader) maps = np.zeros(nc) # mAP per class - results = (0, 0, 0, 0, 0) # P, R, mAP, F1, test_loss + results = (0, 0, 0, 0, 0, 0, 0) # P, R, mAP, F1, test_loss # n_burnin = min(round(nb / 5 + 1), 1000) # burn-in batches t0 = time.time() for epoch in range(start_epoch, epochs): @@ -341,7 +341,7 @@ if __name__ == '__main__': parser.add_argument('--cfg', type=str, default='cfg/yolov3-spp.cfg', help='cfg file path') parser.add_argument('--data', type=str, default='data/coco.data', help='coco.data file path') parser.add_argument('--multi-scale', action='store_true', help='train at (1/1.5)x - 1.5x sizes') - parser.add_argument('--img-size', type=int, default=416, help='inference size (pixels)') + parser.add_argument('--img-size', type=int, default=320, help='inference size (pixels)') parser.add_argument('--rect', action='store_true', help='rectangular training') parser.add_argument('--resume', action='store_true', help='resume training flag') parser.add_argument('--transfer', action='store_true', help='transfer learning flag')