Skip to content
Snippets Groups Projects
Commit 44ff9983 authored by l-ilya's avatar l-ilya Committed by Kai Chen
Browse files

fixed misspell in the "filename" key of img_info (#1107)

parent 009a82c2
No related branches found
No related tags found
No related merge requests found
...@@ -239,7 +239,7 @@ class CustomDataset(Dataset): ...@@ -239,7 +239,7 @@ class CustomDataset(Dataset):
if self.with_seg: if self.with_seg:
gt_seg = mmcv.imread( gt_seg = mmcv.imread(
osp.join(self.seg_prefix, osp.join(self.seg_prefix,
img_info['file_name'].replace('jpg', 'png')), img_info['filename'].replace('jpg', 'png')),
flag='unchanged') flag='unchanged')
gt_seg = self.seg_transform(gt_seg.squeeze(), img_scale, flip) gt_seg = self.seg_transform(gt_seg.squeeze(), img_scale, flip)
gt_seg = mmcv.imrescale( gt_seg = mmcv.imrescale(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment