res variable

Author: aditya1503

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,