44 lines
1.5 KiB
Bash
Executable File
44 lines
1.5 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
# New VM
|
|
sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3
|
|
bash yolov3/data/get_coco_dataset.sh
|
|
sudo rm -rf cocoapi && git clone https://github.com/cocodataset/cocoapi && cd cocoapi/PythonAPI && make && cd ../.. && cp -r cocoapi/PythonAPI/pycocotools yolov3
|
|
|
|
# Start
|
|
python3 train.py
|
|
|
|
# Resume
|
|
python3 train.py --resume
|
|
|
|
# Detect
|
|
gsutil cp gs://ultralytics/yolov3.pt yolov3/weights
|
|
python3 detect.py
|
|
|
|
# Test
|
|
sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3
|
|
sudo rm -rf cocoapi && git clone https://github.com/cocodataset/cocoapi && cd cocoapi/PythonAPI && make && cd ../.. && cp -r cocoapi/PythonAPI/pycocotools yolov3
|
|
cd yolov3 && python3 test.py --save-json --conf-thres 0.005
|
|
|
|
# Test Darknet
|
|
python3 test.py --img_size 416 --weights ../darknet/backup/yolov3.backup
|
|
|
|
# Download and Resume
|
|
sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 && cd yolov3
|
|
wget https://storage.googleapis.com/ultralytics/yolov3.pt -O weights/latest.pt
|
|
python3 train.py --img_size 416 --batch_size 16 --epochs 1 --resume
|
|
python3 test.py --img_size 416 --weights weights/latest.pt --conf_thres 0.5
|
|
|
|
# Copy latest.pt to bucket
|
|
gsutil cp yolov3/weights/latest.pt gs://ultralytics
|
|
|
|
# Copy latest.pt from bucket
|
|
gsutil cp gs://ultralytics/latest.pt yolov3/weights/latest.pt
|
|
wget https://storage.googleapis.com/ultralytics/latest.pt
|
|
|
|
# Testing
|
|
sudo rm -rf yolov3 && git clone https://github.com/ultralytics/yolov3 && cd yolov3
|
|
python3 train.py --epochs 3 --var 64
|
|
sudo shutdown
|
|
|