From b034382c8bd4a33f4150cb38e87f3e4c218703b0 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 18 May 2019 23:24:26 +0200 Subject: [PATCH] updates --- train.py | 2 +- utils/gcp.sh | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/train.py b/train.py index 1eea7a24..d31ffd26 100644 --- a/train.py +++ b/train.py @@ -138,7 +138,7 @@ def train( # plt.savefig('LR.png', dpi=300) # Dataset - dataset = LoadImagesAndLabels(train_path, img_size, batch_size, augment=False, rect=False, image_weights=False) + dataset = LoadImagesAndLabels(train_path, img_size, batch_size, augment=True, rect=False, image_weights=False) # Initialize distributed training if torch.cuda.device_count() > 1: diff --git a/utils/gcp.sh b/utils/gcp.sh index a7e95bf3..b5109df4 100755 --- a/utils/gcp.sh +++ b/utils/gcp.sh @@ -61,8 +61,10 @@ sudo shutdown # Unit tests python3 detect.py # detect 2 persons, 1 tie -python3 test.py --data data/coco_32img.data # test mAP = 0.78 -python3 train.py --data data/coco_32img.data --epochs 4 --nosave # train 4 epochs +python3 test.py --data data/coco_32img.data # test mAP = 0.8 +python3 train.py --data data/coco_32img.data --epochs 5 --nosave # train 5 epochs +python3 train.py --data data/coco_1cls.data --epochs 5 --nosave # train 5 epochs +python3 train.py --data data/coco_1img.data --epochs 5 --nosave # train 5 epochs # AlexyAB Darknet ./darknet detector train ../supermarket2/supermarket2.data cfg/yolov3-spp-sm2.cfg darknet53.conv.74 # train