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

44ff9983 · l-ilya · Kai Chen · 009a82c2 · 44ff9983
Commit 44ff9983 authored Aug 02, 2019 by l-ilya Committed by Kai Chen Aug 02, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

mmdet/datasets/custom.py mmdet/datasets/custom.py +1 -1

No files found.
--- a/mmdet/datasets/custom.py
+++ b/mmdet/datasets/custom.py
@@ -239,7 +239,7 @@ class CustomDataset(Dataset):
        if self.with_seg:
            gt_seg = mmcv.imread(
                osp.join(self.seg_prefix,
-                         img_info['file_name'].replace('jpg', 'png')),
+                         img_info['filename'].replace('jpg', 'png')),
                flag='unchanged')
            gt_seg = self.seg_transform(gt_seg.squeeze(), img_scale, flip)
            gt_seg = mmcv.imrescale(