zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

weights

This commit is contained in:
Jirka 2020-07-16 11:45:17 +02:00
parent 3bf77ca180
commit f0d1e099e2
3 changed files with 16 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
.github/workflows/ci-testing.yml vendored
View File

@ -47,28 +47,29 @@ jobs:
pip list pip list
shell: bash shell: bash
#- name: Download data - name: Download data
# run: | run: |
# python3 -c "from utils.google_utils import *; gdrive_download('1n_oKgR81BJtqk75b00eAjdv03qVCQn2f', 'coco128.zip')" && mv ./coco128 ../ python -c "from utils.google_utils import * ; gdrive_download('1n_oKgR81BJtqk75b00eAjdv03qVCQn2f', 'coco128.zip')"
bash weights/download_weights.sh
- name: Tests - name: Tests
run: | run: |
for x in yolov5s yolov5m yolov5l yolov5x # models for name in yolov5s yolov5m yolov5l yolov5x # models
do do
di=cpu # inference devices di=cpu # inference devices
# train # train
python train.py --weights $x.pt --cfg $x.yaml --epochs 3 --img 320 --device $di --batch-size 2 python train.py --weights weights/$name.pt --cfg models/$name.yaml --epochs 1 --img 320 --device $di --batch-size 2
# detect official # detect official
python detect.py --weights $x.pt --device $di --batch-size 2 python detect.py --weights weights/$name.pt --device $di --batch-size 2
# detect custom # detect custom
python detect.py --weights runs/exp0/weights/last.pt --device $di --batch-size 2 python detect.py --weights runs/exp0/weights/last.pt --device $di --batch-size 2
# test official # test official
python test.py --weights $x.pt --device $di --batch-size 2 python test.py --weights weights/$name.pt --device $di --batch-size 2
# test custom # test custom
python test.py --weights runs/exp0/weights/last.pt --device $di --batch-size 2 python test.py --weights runs/exp0/weights/last.pt --device $di --batch-size 2
# inspect # inspect
python models/yolo.py --cfg $x.yaml python models/yolo.py --cfg models/$name.yaml
# export # export
python models/export.py --weights $x.pt --img 640 --batch 1 python models/export.py --weights weights/$name.pt --img 640 --batch 1
done done
shell: bash shell: bash

4
data/coco128.yaml
View File

@ -8,8 +8,8 @@
# train and val data as 1) directory: path/images/, 2) file: path/images.txt, or 3) list: [path1/images/, path2/images/] # train and val data as 1) directory: path/images/, 2) file: path/images.txt, or 3) list: [path1/images/, path2/images/]
train: ../coco128/images/train2017/ # 128 images train: coco128/images/train2017/ # 128 images
val: ../coco128/images/train2017/ # 128 images val: coco128/images/train2017/ # 128 images
# number of classes # number of classes
nc: 80 nc: 80

6
weights/download_weights.sh
View File

@ -1,8 +1,10 @@
#!/bin/bash #!/bin/bash
# Download common models # Download common models
python3 -c "from utils.google_utils import *; python -c "
from utils.google_utils import *;
attempt_download('weights/yolov5s.pt'); attempt_download('weights/yolov5s.pt');
attempt_download('weights/yolov5m.pt'); attempt_download('weights/yolov5m.pt');
attempt_download('weights/yolov5l.pt'); attempt_download('weights/yolov5l.pt');
attempt_download('weights/yolov5x.pt')" attempt_download('weights/yolov5x.pt')
"