apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

clean up train.py

Browse Source
This commit is contained in:
Glenn Jocher
2018-10-09 19:22:33 +02:00
parent b7d039737a
commit 0cc8f2be01
3 changed files with 25 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/gcp.sh
+7 -6
View File
@@ -11,21 +11,22 @@ gsutil cp gs://ultralytics/fresh9_5_e201.pt yolov3/checkpoints
python3 detect.py
# 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_thres 0.5
# Download and Test
sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3
cd yolov3
cd checkpoints
cd yolov3/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
cd yolov3/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
python3 train.py -img_size 416 -batch_size 12 -epochs 1 -resume 1
python3 test.py -img_size 416 -weights_path checkpoints/latest.pt -conf_thres 0.5