res variable

cleanlab · Jan 29, 2024 · 704a473 · 704a473
1 parent 4a2e1da
commit 704a473
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/object_detection/detectron2_training-kfold.ipynb b/object_detection/detectron2_training-kfold.ipynb
@@ -359,7 +359,7 @@
     "        im = cv2.imread(im_name)\n",
     "        outputs = predictor(im)\n",
     "        result_dict[im_name] = (format_detectron2_predictions(outputs[\"instances\"].to(\"cpu\"),cfg.MODEL.ROI_HEADS.NUM_CLASSES))\n",
-    "    pickle.dump(result_dict,open(\"train_results_fold_\"+str(k)+\".pkl\",'wb'))"
+    "    pickle.dump(result_dict,open(\"results_fold_\"+str(k)+\".pkl\",'wb'))"
    ]
   },
   {
@@ -374,7 +374,7 @@
     "for k in range(0,NUM_FOLDS):\n",
     "    res_d = pickle.load(open(\"results_fold_\"+str(k)+'.pkl','rb'))\n",
     "    for r in res_d:\n",
-    "        result_dict[r] = res_d[i]"
+    "        result_dict[r] = res_d[r]"
    ]
   },
   {
@@ -410,7 +410,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.12"
+   "version": "3.11.5"
   }
  },
  "nbformat": 4,