Commit
·
850f98f
1
Parent(s):
27a4736
Created using Colaboratory
Browse files- tutorial.ipynb +4 -4
tutorial.ipynb
CHANGED
@@ -957,7 +957,6 @@
|
|
957 |
"# Unit tests\n",
|
958 |
"%%shell\n",
|
959 |
"export PYTHONPATH=\"$PWD\" # to run *.py. files in subdirectories\n",
|
960 |
-
"\n",
|
961 |
"rm -rf runs # remove runs/\n",
|
962 |
"for m in yolov5s; do # models\n",
|
963 |
" python train.py --weights $m.pt --epochs 3 --img 320 --device 0 # train pretrained\n",
|
@@ -968,9 +967,10 @@
|
|
968 |
" python val.py --weights $m.pt --device $d # val official\n",
|
969 |
" python val.py --weights runs/train/exp/weights/best.pt --device $d # val custom\n",
|
970 |
" done\n",
|
971 |
-
"
|
972 |
-
"
|
973 |
-
"
|
|
|
974 |
"done"
|
975 |
],
|
976 |
"execution_count": null,
|
|
|
957 |
"# Unit tests\n",
|
958 |
"%%shell\n",
|
959 |
"export PYTHONPATH=\"$PWD\" # to run *.py. files in subdirectories\n",
|
|
|
960 |
"rm -rf runs # remove runs/\n",
|
961 |
"for m in yolov5s; do # models\n",
|
962 |
" python train.py --weights $m.pt --epochs 3 --img 320 --device 0 # train pretrained\n",
|
|
|
967 |
" python val.py --weights $m.pt --device $d # val official\n",
|
968 |
" python val.py --weights runs/train/exp/weights/best.pt --device $d # val custom\n",
|
969 |
" done\n",
|
970 |
+
"python hubconf.py # hub\n",
|
971 |
+
"python models/yolo.py --cfg $m.yaml # build PyTorch model\n",
|
972 |
+
"python models/tf.py --weights $m.pt # build TensorFlow model\n",
|
973 |
+
"python export.py --img 128 --batch 1 --weights $m.pt --include torchscript onnx # export\n",
|
974 |
"done"
|
975 |
],
|
976 |
"execution_count": null,
|