From ef1703f2b83811c3b48e51cb510843c45f3c06da Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 12 Jun 2019 13:19:17 +0200 Subject: [PATCH] updates --- train.py | 2 +- utils/gcp.sh | 21 ++++++++++++++++++--- 2 files changed, 19 insertions(+), 4 deletions(-) diff --git a/train.py b/train.py index e1f5758d..f3cf9380 100644 --- a/train.py +++ b/train.py @@ -213,7 +213,7 @@ def train( # Plot images with bounding boxes if epoch == 0 and i == 0: - plot_images(imgs=imgs, targets=targets, fname='train_batch0.jpg') + plot_images(imgs=imgs, targets=targets, fname='train_batch%g.jpg' % i) # SGD burn-in if epoch == 0 and i <= n_burnin: diff --git a/utils/gcp.sh b/utils/gcp.sh index bbd2a112..2d35e2b2 100755 --- a/utils/gcp.sh +++ b/utils/gcp.sh @@ -67,18 +67,33 @@ python3 train.py --data data/coco_1cls.data --epochs 5 --nosave # train 5 epoch python3 train.py --data data/coco_1img.data --epochs 5 --nosave # train 5 epochs # AlexyAB Darknet -gsutil cp -r gs://sm4/supermarket2 . # dataset from bucket +gsutil cp -r gs://sm6/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 calc_anchors data/coco_img64.data -num_of_clusters 9 -width 320 -height 320 # kmeans anchor calculation -./darknet detector train ../supermarket2/supermarket2.data ../yolov3-spp-sm2-1cls-kmeans.cfg darknet53.conv.74 -map -dont_show # train spp +./darknet detector train ../supermarket2/supermarket2.data ../yolo_v3_spp_pan_scale.cfg darknet53.conv.74 -map -dont_show # train spp ./darknet detector train ../yolov3/data/coco.data ../yolov3-spp.cfg darknet53.conv.74 -map -dont_show # train spp coco +./darknet detector train ../supermarket2/supermarket2.data ../yolov3-spp-sm2-1cls-scalexy_variable.cfg darknet53.conv.74 -map -dont_show # train spp +gsutil cp -r backup/*5000.weights gs://sm6/weights +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 --evolve # train ultralytics +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 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 +python3 test.py --data ../supermarket2/supermarket2.data --weights weights/yolov3-spp-sm2-1cls_5000.weights --cfg ../yolov3-spp-sm2-1cls.cfg --img-size 320 --conf-thres 0.2 # test +python3 test.py --data ../supermarket2/supermarket2.data --weights weights/yolov3-spp-sm2-1cls-scalexy_125_5000.weights --cfg ../yolov3-spp-sm2-1cls-scalexy_125.cfg --img-size 320 --conf-thres 0.2 # test +python3 test.py --data ../supermarket2/supermarket2.data --weights weights/yolov3-spp-sm2-1cls-scalexy_150_5000.weights --cfg ../yolov3-spp-sm2-1cls-scalexy_150.cfg --img-size 320 --conf-thres 0.2 # test +python3 test.py --data ../supermarket2/supermarket2.data --weights weights/yolov3-spp-sm2-1cls-scalexy_200_5000.weights --cfg ../yolov3-spp-sm2-1cls-scalexy_200.cfg --img-size 320 --conf-thres 0.2 # test +python3 test.py --data ../supermarket2/supermarket2.data --weights ../darknet/backup/yolov3-spp-sm2-1cls-scalexy_variable_5000.weights --cfg ../yolov3-spp-sm2-1cls-scalexy_variable.cfg --img-size 320 --conf-thres 0.2 # test + + + + + # Debug/Development python3 train.py --evolve --data data/coco_1k5k.data --epochs 30 --img-size 320 gsutil cp evolve.txt gs://ultralytics