Merge pull request #86 from johnnylu305/master

Update voc.py
kazuto1011 · Aug 2, 2020 · 4219467 · 4219467
2 parents 4b71ed4 + 380b11d
commit 4219467
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/datasets/voc.py b/libs/datasets/voc.py
@@ -45,8 +45,8 @@ def _set_files(self):
     def _load_data(self, index):
         # Set paths
         image_id = self.files[index]
-        image_path = osp.join(self.root, self.image_dir, image_id + ".jpg")
-        label_path = osp.join(self.root, self.label_dir, image_id + ".png")
+        image_path = osp.join(self.image_dir, image_id + ".jpg")
+        label_path = osp.join(self.label_dir, image_id + ".png")
         # Load an image
         image = cv2.imread(image_path, cv2.IMREAD_COLOR).astype(np.float32)
         label = np.asarray(Image.open(label_path), dtype=np.int32)