From e67cee4a0c21d89ed67fcea4681a939f23fa13f4 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 23 May 2019 12:27:46 +0200 Subject: [PATCH] updates --- train.py | 20 ++++++++++---------- utils/gcp.sh | 2 +- 2 files changed, 11 insertions(+), 11 deletions(-) diff --git a/train.py b/train.py index 052aaf11..5b138f61 100644 --- a/train.py +++ b/train.py @@ -11,15 +11,15 @@ from utils.datasets import * from utils.utils import * # Hyperparameters: train.py --evolve --epochs 2 --img-size 320, Metrics: 0.204 0.302 0.175 0.234 (square smart) -hyp = {'xy': 0.167, # xy loss gain - 'wh': 0.09339, # wh loss gain - 'cls': 0.03868, # cls loss gain - 'conf': 4.546, # conf loss gain - 'iou_t': 0.2454, # iou target-anchor training threshold - 'lr0': 0.000198, # initial learning rate - 'lrf': -5., # final learning rate = lr0 * (10 ** lrf) - 'momentum': 0.95, # SGD momentum - 'weight_decay': 0.0007838} # optimizer weight decay +hyp = {'xy': 0.2, # xy loss gain + 'wh': 0.1, # wh loss gain + 'cls': 0.04, # cls loss gain + 'conf': 4.5, # conf loss gain + 'iou_t': 0.5, # iou target-anchor training threshold + 'lr0': 0.001, # initial learning rate + 'lrf': -4., # final learning rate = lr0 * (10 ** lrf) + 'momentum': 0.90, # SGD momentum + 'weight_decay': 0.0005} # optimizer weight decay # Hyperparameters: Original, Metrics: 0.172 0.304 0.156 0.205 (square) @@ -310,7 +310,7 @@ if __name__ == '__main__': parser.add_argument('--cfg', type=str, default='cfg/yolov3-spp.cfg', help='cfg file path') parser.add_argument('--data-cfg', type=str, default='data/coco_32img.data', help='coco.data file path') parser.add_argument('--multi-scale', action='store_true', help='random image sizes per batch 320 - 608') - parser.add_argument('--img-size', type=int, default=320, help='inference size (pixels)') + parser.add_argument('--img-size', type=int, default=416, help='inference size (pixels)') parser.add_argument('--resume', action='store_true', help='resume training flag') parser.add_argument('--transfer', action='store_true', help='transfer learning flag') parser.add_argument('--num-workers', type=int, default=4, help='number of Pytorch DataLoader workers') diff --git a/utils/gcp.sh b/utils/gcp.sh index b6af54c4..b9028b05 100755 --- a/utils/gcp.sh +++ b/utils/gcp.sh @@ -71,7 +71,7 @@ gsutil cp -r gs://sm4/supermarket2 . # dataset from bucket rm -rf darknet && git clone https://github.com/AlexeyAB/darknet && cd darknet && wget -c https://pjreddie.com/media/files/darknet53.conv.74 # sudo apt install libopencv-dev && make ./darknet detector train ../supermarket2/supermarket2.data cfg/yolov3-spp-sm2-1cls.cfg darknet53.conv.74 -map -dont_show # train ./darknet detector train ../supermarket2/supermarket2.data cfg/yolov3-spp-sm2-1cls.cfg backup/yolov3-spp-sm2-1cls_last.weights # resume -python3 train.py --data ../supermarket2/supermarket2.data --cfg cfg/yolov3-spp-sm2-1cls.cfg # test +python3 train.py --data ../supermarket2/supermarket2.data --cfg cfg/yolov3-spp-sm2-1cls.cfg --epochs 100 --num-workers 8 --img-size 320 --nosave # train ultralytics python3 test.py --data ../supermarket2/supermarket2.data --weights ../darknet/backup/yolov3-spp-sm2-1cls_5000.weights --cfg cfg/yolov3-spp-sm2-1cls.cfg # test gsutil cp -r backup/*.weights gs://sm4/weights # weights to bucket