2018-08-26 08:51:39 +00:00
|
|
|
#!/usr/bin/env bash
|
|
|
|
|
|
2019-02-20 22:21:42 +00:00
|
|
|
# New VM
|
2019-03-25 18:39:14 +00:00
|
|
|
rm -rf yolov3 weights coco
|
2019-03-25 17:35:39 +00:00
|
|
|
git clone https://github.com/ultralytics/yolov3
|
2019-03-25 18:39:14 +00:00
|
|
|
bash yolov3/weights/download_yolov3_weights.sh && cp -r weights yolov3
|
2019-02-20 22:21:42 +00:00
|
|
|
bash yolov3/data/get_coco_dataset.sh
|
2019-03-25 17:35:39 +00:00
|
|
|
git clone https://github.com/cocodataset/cocoapi && cd cocoapi/PythonAPI && make && cd ../.. && cp -r cocoapi/PythonAPI/pycocotools yolov3
|
2019-04-24 15:02:52 +00:00
|
|
|
sudo shutdown
|
2019-02-20 22:21:42 +00:00
|
|
|
|
2019-03-25 18:39:14 +00:00
|
|
|
# Re-clone
|
2019-04-19 11:18:47 +00:00
|
|
|
rm -rf yolov3
|
2019-04-11 10:47:35 +00:00
|
|
|
git clone https://github.com/ultralytics/yolov3 # master
|
2019-04-24 12:19:43 +00:00
|
|
|
# git clone -b test --depth 1 https://github.com/ultralytics/yolov3 test # branch
|
2019-03-25 18:39:14 +00:00
|
|
|
cp -r cocoapi/PythonAPI/pycocotools yolov3
|
2019-04-23 16:54:27 +00:00
|
|
|
cp -r weights yolov3
|
2019-03-25 18:39:14 +00:00
|
|
|
cd yolov3
|
|
|
|
|
|
|
|
|
|
# Train
|
|
|
|
|
python3 train.py
|
2018-08-26 08:51:39 +00:00
|
|
|
|
|
|
|
|
# Resume
|
2018-12-05 13:34:53 +00:00
|
|
|
python3 train.py --resume
|
2018-08-26 08:51:39 +00:00
|
|
|
|
|
|
|
|
# Detect
|
2019-03-22 15:53:06 +00:00
|
|
|
python3 detect.py
|
2018-08-26 08:51:39 +00:00
|
|
|
|
2018-09-04 12:36:51 +00:00
|
|
|
# Test
|
2019-03-30 17:45:04 +00:00
|
|
|
python3 test.py --save-json
|
2018-09-10 14:41:02 +00:00
|
|
|
|
2019-03-25 18:39:14 +00:00
|
|
|
# Git pull
|
|
|
|
|
git pull https://github.com/ultralytics/yolov3 # master
|
2019-04-06 14:16:40 +00:00
|
|
|
git pull https://github.com/ultralytics/yolov3 test # branch
|
2018-12-04 18:17:03 +00:00
|
|
|
|
2019-03-25 18:39:14 +00:00
|
|
|
# Test Darknet training
|
|
|
|
|
python3 test.py --weights ../darknet/backup/yolov3.backup
|
2018-10-09 17:22:33 +00:00
|
|
|
|
2019-03-25 18:39:14 +00:00
|
|
|
# Copy latest.pt TO bucket
|
2019-03-25 13:59:38 +00:00
|
|
|
gsutil cp yolov3/weights/latest1gpu.pt gs://ultralytics
|
2018-11-13 11:20:01 +00:00
|
|
|
|
2019-03-25 18:39:14 +00:00
|
|
|
# Copy latest.pt FROM bucket
|
2018-11-13 11:20:01 +00:00
|
|
|
gsutil cp gs://ultralytics/latest.pt yolov3/weights/latest.pt
|
2019-03-30 17:45:04 +00:00
|
|
|
wget https://storage.googleapis.com/ultralytics/yolov3/latest_v1_0.pt -O weights/latest_v1_0.pt
|
|
|
|
|
wget https://storage.googleapis.com/ultralytics/yolov3/best_v1_0.pt -O weights/best_v1_0.pt
|
2018-11-13 11:20:01 +00:00
|
|
|
|
2019-04-18 14:07:47 +00:00
|
|
|
# Reproduce tutorials
|
2019-04-16 10:49:34 +00:00
|
|
|
rm results*.txt # WARNING: removes existing results
|
2019-04-16 21:57:43 +00:00
|
|
|
python3 train.py --nosave --data data/coco_1img.data && mv results.txt results3_1img.txt
|
|
|
|
|
python3 train.py --nosave --data data/coco_10img.data && mv results.txt results3_10img.txt
|
2019-04-17 13:52:51 +00:00
|
|
|
python3 train.py --nosave --data data/coco_100img.data && mv results.txt results4_100img.txt
|
2019-04-16 21:57:43 +00:00
|
|
|
python3 train.py --nosave --data data/coco_100img.data --transfer && mv results.txt results3_100imgTL.txt
|
2019-04-16 10:49:34 +00:00
|
|
|
python3 -c "from utils import utils; utils.plot_results()"
|
|
|
|
|
gsutil cp results*.txt gs://ultralytics
|
|
|
|
|
gsutil cp results.png gs://ultralytics
|
|
|
|
|
sudo shutdown
|
2019-04-02 16:04:04 +00:00
|
|
|
|
2019-04-23 16:53:36 +00:00
|
|
|
# Reproduce mAP
|
|
|
|
|
python3 test.py --save-json --img-size 608 --batch-size 16
|
|
|
|
|
python3 test.py --save-json --img-size 416
|
|
|
|
|
python3 test.py --save-json --img-size 320
|
|
|
|
|
sudo shutdown
|
|
|
|
|
|
2019-04-19 11:18:47 +00:00
|
|
|
# Unit tests
|
|
|
|
|
rm -rf yolov3
|
|
|
|
|
git clone https://github.com/ultralytics/yolov3 # master
|
2019-04-23 16:53:36 +00:00
|
|
|
cp -r cocoapi/PythonAPI/pycocotools yolov3
|
2019-04-24 19:37:32 +00:00
|
|
|
cp -r weights yolov3 && cd yolov3
|
2019-04-24 19:41:18 +00:00
|
|
|
python3 detect.py # detect 2 persons, 1 tie
|
|
|
|
|
python3 test.py --data data/coco_32img.data # test mAP = 0.78
|
|
|
|
|
python3 train.py --data data/coco_32img.data --epochs 4 --nosave # train 4 epochs
|
2019-04-19 11:18:47 +00:00
|
|
|
|
2019-04-18 14:07:47 +00:00
|
|
|
# Debug/Development
|
2019-04-19 11:18:47 +00:00
|
|
|
rm -rf yolov3
|
2019-04-18 14:07:47 +00:00
|
|
|
git clone https://github.com/ultralytics/yolov3 # master
|
|
|
|
|
# git clone -b test --depth 1 https://github.com/ultralytics/yolov3 yolov3_test # branch
|
|
|
|
|
cp -r cocoapi/PythonAPI/pycocotools yolov3
|
2019-04-19 11:18:47 +00:00
|
|
|
cp -r weights yolov3 && cd yolov3
|
2019-04-19 11:24:49 +00:00
|
|
|
python3 train.py --evolve --data data/coco_100img.data --num-workers 2 --epochs 30
|
2019-04-18 15:06:09 +00:00
|
|
|
gsutil cp evolve.txt gs://ultralytics
|
|
|
|
|
sudo shutdown
|
2019-04-23 16:53:36 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
