From e81c1ab501aa676f6d95826e924aeda7a83ce11f Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 12 Jun 2019 13:57:32 +0200 Subject: [PATCH] updates --- utils/datasets.py | 2 +- utils/gcp.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/datasets.py b/utils/datasets.py index 98d855e0..48f18dbf 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -269,7 +269,7 @@ class LoadImagesAndLabels(Dataset): # for training/testing # Augment image and labels if self.augment: - img, labels = random_affine(img, labels, degrees=(-5, 5), translate=(0.10, 0.10), scale=(0.90, 1.10)) + img, labels = random_affine(img, labels, degrees=(-5, 5), translate=(0.0, 0.0), scale=(1.0, 1.0)) nL = len(labels) # number of labels if nL: diff --git a/utils/gcp.sh b/utils/gcp.sh index 2d35e2b2..5a722a55 100755 --- a/utils/gcp.sh +++ b/utils/gcp.sh @@ -80,7 +80,7 @@ sudo shutdown ./darknet detector train ../supermarket2/supermarket2.data ../yolov3-tiny-sm2-1cls.cfg yolov3-tiny.conv.15 -map -dont_show # train tiny ./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 --epochs 100 --num-workers 8 --img-size 320 # train ultralytics +python3 train.py --data ../supermarket2/supermarket2.data --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://sm6/weights # weights to bucket