updates
This commit is contained in:
parent
8a87521044
commit
c01b8e6b7c
2
train.py
2
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')
|
||||
|
||||
|
|
10
utils/gcp.sh
10
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
|
||||
|
|
Loading…
Reference in New Issue