Yolo update

Author: adipandas

tracking-yolo-model.ipynb

@@ -136,8 +136,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "yolomodel = {\"config_path\":\"/home/adi/computer_vision/multi-object-tracker/yolo_dir/yolov3.cfg\",\n",
-    "              \"model_weights_path\":\"/home/adi/computer_vision/multi-object-tracker/yolo_dir/yolov3.weights\",\n",
+    "yolomodel = {\"config_path\":\"./yolo_dir/yolov3.cfg\",\n",
+    "              \"model_weights_path\":\"./yolo_dir/yolov3.weights\",\n",
     "              \"coco_names\":\"./yolo_dir/coco.names\",\n",
     "              \"confidence_threshold\": 0.5,\n",
     "              \"threshold\":0.3\n",
@@ -201,11 +201,11 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": 8,
    "metadata": {},
    "outputs": [],
    "source": [
-    "video_src = \"./video_test2.mp4\"#0\n",
+    "video_src = \"./data/video_test2.mp4\"#0\n",
     "cap = cv.VideoCapture(video_src)"
    ]
   },
@@ -218,9 +218,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 9,
    "metadata": {
-    "scrolled": true
+    "scrolled": false
    },
    "outputs": [
     {
@@ -287,19 +287,18 @@
     "                    0.5, (0, 255, 0), 2)\n",
     "        cv.circle(image, (centroid[0], centroid[1]), 4, (0, 255, 0), -1)\n",
     "        \n",
-    "    #cv.imshow(\"image\", image)\n",
+    "    cv.imshow(\"image\", image)\n",
     "    \n",
     "    if cv.waitKey(1) & 0xFF == ord('q'):\n",
     "        break\n",
     "        \n",
     "    if writer is None:\n",
-    "        # initialize our video writer\n",
     "        fourcc = cv.VideoWriter_fourcc(*\"MJPG\")\n",
     "        writer = cv.VideoWriter(\"output.avi\", fourcc, 30, (W, H), True)\n",
     "    writer.write(image)\n",
     "writer.release()\n",
     "cap.release()\n",
-    "#cv.destroyWindow(\"image\")"
+    "cv.destroyWindow(\"image\")"
    ]
   },
   {