diff --git a/train.py b/train.py index 421e256a..7d28e0e9 100644 --- a/train.py +++ b/train.py @@ -11,8 +11,7 @@ from utils.datasets import * from utils.utils import * # Hyperparameters -# 0.852 0.94 0.924 0.883 1.33 # results -# 8.52 0.06833 0.01524 0.01509 0.9013 0.1003 0.001325 -3.853 0.8948 0.0004053 # hyp +# 0.852 0.94 0.924 0.883 1.33 8.52 0.06833 0.01524 0.01509 0.9013 0.1003 0.001325 -3.853 0.8948 0.0004053 # hyp hyp = {'k': 8.52, # loss multiple 'xy': 0.06833, # xy loss fraction 'wh': 0.01524, # wh loss fraction diff --git a/weights/gcp.sh b/weights/gcp.sh index 060e13b7..d231a4f4 100755 --- a/weights/gcp.sh +++ b/weights/gcp.sh @@ -9,7 +9,7 @@ git clone https://github.com/cocodataset/cocoapi && cd cocoapi/PythonAPI && make sudo reboot now # Re-clone -sudo rm -rf yolov3 +rm -rf yolov3 git clone https://github.com/ultralytics/yolov3 # master # git clone -b test --depth 1 https://github.com/ultralytics/yolov3 yolov3_test # branch cp -r weights yolov3 @@ -54,17 +54,27 @@ gsutil cp results*.txt gs://ultralytics gsutil cp results.png gs://ultralytics sudo shutdown +# Unit tests +rm -rf yolov3 +git clone https://github.com/ultralytics/yolov3 # master +cp -r weights yolov3 && cd yolov3 +python3 detecty.py # detect +python3 test.py --data data/coco_32img.data # test +python3 train.py --data data/coco_32img.data --epochs 3 --nosave # train + # Debug/Development -sudo rm -rf yolov3 +rm -rf yolov3 git clone https://github.com/ultralytics/yolov3 # master # git clone -b test --depth 1 https://github.com/ultralytics/yolov3 yolov3_test # branch -cp -r weights yolov3 cp -r cocoapi/PythonAPI/pycocotools yolov3 -cd yolov3 -python3 train.py --evolve --data data/coco_100img.data --num-workers 2 +cp -r weights yolov3 && cd yolov3 +python3 train.py --nosave --data data/coco_100img.data --num-workers 2 gsutil cp evolve.txt gs://ultralytics sudo shutdown +gsutil cp results*.txt gs://ultralytics +sudo shutdown now +