car-detection-bayes/utils/gcp.sh

130 lines
7.2 KiB
Bash
Raw Normal View History

2018-08-26 08:51:39 +00:00
#!/usr/bin/env bash
2019-02-20 22:21:42 +00:00
# New VM
2019-08-04 18:03:46 +00:00
rm -rf sample_data yolov3 darknet apex coco cocoapi knife knifec
2019-03-25 17:35:39 +00:00
git clone https://github.com/ultralytics/yolov3
# git clone https://github.com/AlexeyAB/darknet && cd darknet && make GPU=1 CUDNN=1 CUDNN_HALF=1 OPENCV=0 && wget -c https://pjreddie.com/media/files/darknet53.conv.74 && cd ..
git clone https://github.com/NVIDIA/apex && cd apex && pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" . --user && cd .. && rm -rf apex
2019-08-31 11:21:15 +00:00
# git clone https://github.com/cocodataset/cocoapi && cd cocoapi/PythonAPI && make && cd ../.. && cp -r cocoapi/PythonAPI/pycocotools yolov3
2019-08-15 23:31:59 +00:00
sudo conda install -y -c conda-forge scikit-image tensorboard pycocotools
2019-08-04 18:03:46 +00:00
python3 -c "
from yolov3.utils.google_utils import gdrive_download
2019-08-31 11:21:15 +00:00
gdrive_download('1HaXkef9z6y5l4vUnCYgdmEAj61c6bfWO','coco.zip')"
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-05-13 19:18:15 +00:00
rm -rf yolov3 # Warning: remove existing
2019-08-24 14:43:43 +00:00
git clone https://github.com/ultralytics/yolov3 && cd yolov3 # master
2019-04-24 12:19:43 +00:00
# git clone -b test --depth 1 https://github.com/ultralytics/yolov3 test # branch
2019-08-26 12:47:36 +00:00
python3 train.py --img-size 320 --weights weights/darknet53.conv.74 --epochs 27 --batch-size 64 --accumulate 1
2019-03-25 18:39:14 +00:00
# Train
python3 train.py
2018-08-26 08:51:39 +00:00
# Resume
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
python3 test.py --save-json
2018-09-10 14:41:02 +00:00
2019-07-24 18:16:35 +00:00
# Evolve
2019-07-24 18:32:24 +00:00
for i in {0..500}
2019-07-24 18:16:35 +00:00
do
python3 train.py --data data/coco.data --img-size 320 --epochs 1 --batch-size 64 --accumulate 1 --evolve --bucket yolov4
done
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-07-15 15:54:31 +00:00
# Copy last.pt TO bucket
gsutil cp yolov3/weights/last1gpu.pt gs://ultralytics
2018-11-13 11:20:01 +00:00
2019-07-15 15:54:31 +00:00
# Copy last.pt FROM bucket
gsutil cp gs://ultralytics/last.pt yolov3/weights/last.pt
wget https://storage.googleapis.com/ultralytics/yolov3/last_v1_0.pt -O weights/last_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-05-08 11:31:49 +00:00
python3 train.py --nosave --data data/coco_1img.data && mv results.txt results0r_1img.txt
python3 train.py --nosave --data data/coco_10img.data && mv results.txt results0r_10img.txt
python3 train.py --nosave --data data/coco_100img.data && mv results.txt results0r_100img.txt
2019-07-12 13:44:39 +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()"
2019-07-12 13:44:39 +00:00
# gsutil cp results*.txt gs://ultralytics
2019-04-16 10:49:34 +00:00
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
2019-05-02 22:26:26 +00:00
python3 test.py --save-json --img-size 608
2019-04-23 16:53:36 +00:00
python3 test.py --save-json --img-size 416
python3 test.py --save-json --img-size 320
sudo shutdown
2019-08-05 11:50:06 +00:00
# Benchmark script
git clone https://github.com/ultralytics/yolov3 # clone our repo
git clone https://github.com/NVIDIA/apex && cd apex && pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" . --user && cd .. && rm -rf apex # install nvidia apex
python3 -c "from yolov3.utils.google_utils import gdrive_download; gdrive_download('1HaXkef9z6y5l4vUnCYgdmEAj61c6bfWO','coco.zip')" # download coco dataset (20GB)
2019-08-05 12:15:12 +00:00
cd yolov3 && clear && python3 train.py --epochs 1 # run benchmark (~30 min)
2019-08-05 11:50:06 +00:00
2019-04-19 11:18:47 +00:00
# Unit tests
2019-04-24 19:41:18 +00:00
python3 detect.py # detect 2 persons, 1 tie
2019-05-18 21:24:26 +00:00
python3 test.py --data data/coco_32img.data # test mAP = 0.8
python3 train.py --data data/coco_32img.data --epochs 5 --nosave # train 5 epochs
python3 train.py --data data/coco_1cls.data --epochs 5 --nosave # train 5 epochs
python3 train.py --data data/coco_1img.data --epochs 5 --nosave # train 5 epochs
2019-04-19 11:18:47 +00:00
2019-05-13 19:18:15 +00:00
# AlexyAB Darknet
2019-06-12 11:19:17 +00:00
gsutil cp -r gs://sm6/supermarket2 . # dataset from bucket
2019-05-21 13:00:16 +00:00
rm -rf darknet && git clone https://github.com/AlexeyAB/darknet && cd darknet && wget -c https://pjreddie.com/media/files/darknet53.conv.74 # sudo apt install libopencv-dev && make
2019-05-29 23:40:35 +00:00
./darknet detector calc_anchors data/coco_img64.data -num_of_clusters 9 -width 320 -height 320 # kmeans anchor calculation
2019-06-12 11:19:17 +00:00
./darknet detector train ../supermarket2/supermarket2.data ../yolo_v3_spp_pan_scale.cfg darknet53.conv.74 -map -dont_show # train spp
2019-05-29 23:40:35 +00:00
./darknet detector train ../yolov3/data/coco.data ../yolov3-spp.cfg darknet53.conv.74 -map -dont_show # train spp coco
2019-06-21 08:58:12 +00:00
./darknet detector train data/coco.data ../yolov3-spp.cfg darknet53.conv.74 -map -dont_show # train spp
2019-06-12 11:19:17 +00:00
gsutil cp -r backup/*5000.weights gs://sm6/weights
sudo shutdown
2019-05-28 14:14:37 +00:00
./darknet detector train ../supermarket2/supermarket2.data ../yolov3-tiny-sm2-1cls.cfg yolov3-tiny.conv.15 -map -dont_show # train tiny
2019-05-21 13:00:16 +00:00
./darknet detector train ../supermarket2/supermarket2.data cfg/yolov3-spp-sm2-1cls.cfg backup/yolov3-spp-sm2-1cls_last.weights # resume
2019-06-12 11:57:32 +00:00
python3 train.py --data ../supermarket2/supermarket2.data --cfg ../yolov3-spp-sm2-1cls.cfg --epochs 100 --num-workers 8 --img-size 320 --nosave # train ultralytics
2019-05-21 15:37:34 +00:00
python3 test.py --data ../supermarket2/supermarket2.data --weights ../darknet/backup/yolov3-spp-sm2-1cls_5000.weights --cfg cfg/yolov3-spp-sm2-1cls.cfg # test
2019-05-28 14:14:37 +00:00
gsutil cp -r backup/*.weights gs://sm6/weights # weights to bucket
2019-05-13 19:18:15 +00:00
2019-06-12 11:19:17 +00:00
python3 test.py --data ../supermarket2/supermarket2.data --weights weights/yolov3-spp-sm2-1cls_5000.weights --cfg ../yolov3-spp-sm2-1cls.cfg --img-size 320 --conf-thres 0.2 # test
python3 test.py --data ../supermarket2/supermarket2.data --weights weights/yolov3-spp-sm2-1cls-scalexy_125_5000.weights --cfg ../yolov3-spp-sm2-1cls-scalexy_125.cfg --img-size 320 --conf-thres 0.2 # test
python3 test.py --data ../supermarket2/supermarket2.data --weights weights/yolov3-spp-sm2-1cls-scalexy_150_5000.weights --cfg ../yolov3-spp-sm2-1cls-scalexy_150.cfg --img-size 320 --conf-thres 0.2 # test
python3 test.py --data ../supermarket2/supermarket2.data --weights weights/yolov3-spp-sm2-1cls-scalexy_200_5000.weights --cfg ../yolov3-spp-sm2-1cls-scalexy_200.cfg --img-size 320 --conf-thres 0.2 # test
python3 test.py --data ../supermarket2/supermarket2.data --weights ../darknet/backup/yolov3-spp-sm2-1cls-scalexy_variable_5000.weights --cfg ../yolov3-spp-sm2-1cls-scalexy_variable.cfg --img-size 320 --conf-thres 0.2 # test
2019-08-11 22:22:23 +00:00
python3 train.py --img-size 320 --epochs 27 --batch-size 64 --accumulate 1 --nosave --notest && python3 test.py --weights weights/last.pt --img-size 320 --save-json && sudo shutdown
2019-06-12 11:19:17 +00:00
2019-04-18 14:07:47 +00:00
# Debug/Development
2019-06-25 10:21:27 +00:00
python3 train.py --data data/coco.data --img-size 320 --single-scale --batch-size 64 --accumulate 1 --epochs 1 --evolve --giou
2019-07-15 15:54:31 +00:00
python3 test.py --weights weights/last.pt --cfg cfg/yolov3-spp.cfg --img-size 320
2019-06-21 21:11:24 +00:00
2019-04-18 15:06:09 +00:00
gsutil cp evolve.txt gs://ultralytics
sudo shutdown
2019-09-16 21:10:21 +00:00
#Docker
sudo docker kill $(sudo docker ps -q)
sudo docker pull ultralytics/yolov3:v1
sudo nvidia-docker run -it --ipc=host --mount type=bind,source="$(pwd)"/coco,target=/usr/src/coco ultralytics/yolov3:v1
clear
while true
do
python3 train.py --data data/coco.data --img-size 320 --batch-size 64 --accumulate 1 --evolve --epochs 1 --adam --bucket yolov4/adamdefaultpw_coco_1e --device 1
done
python3 train.py --data data/coco.data --img-size 320 --batch-size 64 --accumulate 1 --epochs 1 --adam --device 1 --prebias
while true; do python3 train.py --data data/coco.data --img-size 320 --batch-size 64 --accumulate 1 --evolve --epochs 1 --adam --bucket yolov4/adamdefaultpw_coco_1e; done