We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 4b71ed4 commit 380b11dCopy full SHA for 380b11d
libs/datasets/voc.py
@@ -45,8 +45,8 @@ def _set_files(self):
45
def _load_data(self, index):
46
# Set paths
47
image_id = self.files[index]
48
- image_path = osp.join(self.root, self.image_dir, image_id + ".jpg")
49
- 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")
50
# Load an image
51
image = cv2.imread(image_path, cv2.IMREAD_COLOR).astype(np.float32)
52
label = np.asarray(Image.open(label_path), dtype=np.int32)
0 commit comments