reshape to 5 (#90)

Author: aditya1503

object_detection/detectron2_training.ipynb

@@ -186,7 +186,7 @@
     "    res2 = []\n",
     "    for i in res:\n",
     "        if len(i)==0:\n",
-    "            q = np.array(i,dtype=np.float32).reshape((0,num_classes))\n",
+    "            q = np.array(i,dtype=np.float32).reshape((0,5))\n",
     "        else:\n",
     "            q = np.array(i,dtype=np.float32)\n",
     "        res2.append(q)\n",
@@ -235,7 +235,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.12"
+   "version": "3.11.5"
   }
  },
  "nbformat": 4,