This commit is contained in:
Glenn Jocher 2018-10-05 16:38:59 +02:00
parent 8a87521044
commit c01b8e6b7c
2 changed files with 10 additions and 2 deletions

View File

@ -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')

View File

@ -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