Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -179,7 +179,7 @@ def main(opt):
if best_loss == loss_per_target: if best_loss == loss_per_target:
os.system('cp checkpoints/latest.pt checkpoints/best.pt') os.system('cp checkpoints/latest.pt checkpoints/best.pt')
# Save backup checkpoint # Save backup checkpoints every 5 epochs
if (epoch > 0) & (epoch % 5 == 0): if (epoch > 0) & (epoch % 5 == 0):
os.system('cp checkpoints/latest.pt checkpoints/backup' + str(epoch) + '.pt') os.system('cp checkpoints/latest.pt checkpoints/backup' + str(epoch) + '.pt')

10
utils/gcp.sh
View File

@ -13,7 +13,6 @@ python3 detect.py
# Test # Test
python3 test.py -img_size 416 -weights_path checkpoints/latest.pt -conf_thresh 0.5 python3 test.py -img_size 416 -weights_path checkpoints/latest.pt -conf_thresh 0.5
# Download and Test # Download and Test
sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3
cd yolov3 cd yolov3
@ -21,3 +20,12 @@ cd checkpoints
wget https://pjreddie.com/media/files/yolov3.weights wget https://pjreddie.com/media/files/yolov3.weights
cd .. cd ..
python3 test.py -img_size 416 -weights_path checkpoints/backup5.pt -nms_thres 0.45 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