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

Created using Colaboratory

This commit is contained in:
Glenn Jocher 2021-09-17 11:39:00 +02:00
parent 27a4736e96
commit 850f98f508
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tutorial.ipynb vendored
View File

@ -957,7 +957,6 @@
"# Unit tests\n",
"%%shell\n",
"export PYTHONPATH=\"$PWD\" # to run *.py. files in subdirectories\n",
"\n",
"rm -rf runs # remove runs/\n",
"for m in yolov5s; do # models\n",
" python train.py --weights $m.pt --epochs 3 --img 320 --device 0 # train pretrained\n",
@ -969,8 +968,9 @@
" python val.py --weights runs/train/exp/weights/best.pt --device $d # val custom\n",
" done\n",
"python hubconf.py # hub\n",
" python models/yolo.py --cfg $m.yaml # inspect\n",
" python export.py --weights $m.pt --img 640 --batch 1 # export\n",
"python models/yolo.py --cfg $m.yaml # build PyTorch model\n",
"python models/tf.py --weights $m.pt # build TensorFlow model\n",
"python export.py --img 128 --batch 1 --weights $m.pt --include torchscript onnx # export\n",
"done"
],
"execution_count": null,