From c01b8e6b7cfd13cb73e8b4d60263a23f4f892e4e Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Fri, 5 Oct 2018 16:38:59 +0200 Subject: [PATCH] updates --- train.py | 2 +- utils/gcp.sh | 10 +++++++++- 2 files changed, 10 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index f0320401..82a5188e 100644 --- a/train.py +++ b/train.py @@ -179,7 +179,7 @@ def main(opt): if best_loss == loss_per_target: os.system('cp checkpoints/latest.pt checkpoints/best.pt') - # Save backup checkpoint + # Save backup checkpoints every 5 epochs if (epoch > 0) & (epoch % 5 == 0): os.system('cp checkpoints/latest.pt checkpoints/backup' + str(epoch) + '.pt') diff --git a/utils/gcp.sh b/utils/gcp.sh index a52aff2f..42d5ee7c 100644 --- a/utils/gcp.sh +++ b/utils/gcp.sh @@ -13,7 +13,6 @@ python3 detect.py # Test python3 test.py -img_size 416 -weights_path checkpoints/latest.pt -conf_thresh 0.5 - # Download and Test sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 cd yolov3 @@ -21,3 +20,12 @@ cd checkpoints wget https://pjreddie.com/media/files/yolov3.weights cd .. python3 test.py -img_size 416 -weights_path checkpoints/backup5.pt -nms_thres 0.45 + +# Download and Resume +sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 +cd yolov3 +cd checkpoints +wget https://storage.googleapis.com/ultralytics/yolov3.pt +cp yolov3.pt latest.pt +cd .. +python3 train.py -img_size 416 -epochs 1 -resume 1