exploring data

notebooks/model-training-v1.ipynb

@@ -17,6 +17,39 @@
    "source": [
     "print(\"Hello, World!\")"
    ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 2,
+   "id": "d318d1f0",
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Van: 1111 images\n",
+      "Taxi: 748 images\n",
+      "Bicycle: 1618 images\n",
+      "Bus: 2133 images\n",
+      "Car: 6781 images\n",
+      "Motorcycle: 2986 images\n",
+      "Truck: 2033 images\n",
+      "NonVehicles: 8968 images\n"
+     ]
+    }
+   ],
+   "source": [
+    "import os\n",
+    "\n",
+    "data_dir = '../data/raw/vehicle_classification'\n",
+    "\n",
+    "for class_name in os.listdir(data_dir):\n",
+    "    class_path = os.path.join(data_dir, class_name)\n",
+    "    if os.path.isdir(class_path):\n",
+    "        count = len(os.listdir(class_path))\n",
+    "        print(f\"{class_name}: {count} images\")"
+   ]
   }
  ],
  "metadata": {