Notebook cleanup

PUTvision · Mar 10, 2023 · 9350a7b · 9350a7b
1 parent 4bab44e
commit 9350a7b
Showing 1 changed file with 23 additions and 3 deletions.
diff --git a/tutorials/detection/cars_yolov7/car_detection__prepare_and_train.ipynb b/tutorials/detection/cars_yolov7/car_detection__prepare_and_train.ipynb
@@ -277,7 +277,17 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!python train.py --workers 8 --device 0 --batch-size 2 --data car/car.yaml --img 640 640 --cfg cfg/training/yolov7-tiny.yaml --weights yolov7-tiny.pt --name yolov7-car-detector --hyp data/hyp.scratch.custom.yaml --epochs 15 "
+    "!python train.py \\\n",
+    "    --workers 8 \\\n",
+    "    --device 0 \\\n",
+    "    --batch-size 2 \\\n",
+    "    --data car/car.yaml \\\n",
+    "    --img 640 640 \\\n",
+    "    --cfg cfg/training/yolov7-tiny.yaml \\\n",
+    "    --weights yolov7-tiny.pt \\\n",
+    "    --name yolov7-car-detector \\\n",
+    "    --hyp data/hyp.scratch.custom.yaml \\\n",
+    "    --epochs 15 "
    ]
   },
   {
@@ -300,7 +310,13 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!python test.py  --device 0 --batch-size 2 --data car/car.yaml --img 640  --weights runs/train/yolov7-car-detector/weights/best.pt_xx --name yolov7-car-detector"
+    "!python test.py  \\\n",
+    "    --device 0 \\\n",
+    "    --batch-size 2 \\ \n",
+    "    --data car/car.yaml \\\n",
+    "    --img 640  \\\n",
+    "    --weights runs/train/yolov7-car-detector/weights/best.pt_xx \\\n",
+    "    --name yolov7-car-detector"
    ]
   },
   {
@@ -323,7 +339,11 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "!python export.py --weights runs/train/yolov7-car-detector/weights/best.pt --grid --simplify --img-size 640 640"
+    "!python export.py \\\n",
+    "    --weights runs/train/yolov7-car-detector/weights/best.pt \\\n",
+    "    --grid \\\n",
+    "    --simplify \\\n",
+    "    --img-size 640 640"
    ]
   },
   {