Created using Colaboratory

souravs17031999 · Aug 31, 2020 · 7ce086e · 7ce086e
1 parent c965c6d
commit 7ce086e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Single_test_inference.ipynb b/Single_test_inference.ipynb
@@ -140,7 +140,7 @@
         "    def __getitem__(self, index):\n",
         "        img_name,label = self.df[index]\n",
         "        img_path = os.path.join(self.data_dir, img_name+'.png')\n",
-        "        image = cv2.imread(img_path)\n",
+        "        image = cv2.imread(img_path)  # if getting error, use PIL to load the image here.\n",
         "        if self.transform is not None:\n",
         "            image = self.transform(image)\n",
         "        return image, label"