reshape to 5 (#90)

cleanlab · May 25, 2024 · 67770ef · 67770ef
1 parent b22ad9b
commit 67770ef
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/object_detection/detectron2_training.ipynb b/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,