From 44ff99831a7ed1a4f14dd240b69e74d88d7944f5 Mon Sep 17 00:00:00 2001
From: l-ilya <41329713+l-ilya@users.noreply.github.com>
Date: Fri, 2 Aug 2019 17:58:34 +0300
Subject: [PATCH] fixed misspell in the "filename" key of img_info (#1107)
---
mmdet/datasets/custom.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/mmdet/datasets/custom.py b/mmdet/datasets/custom.py
index 25944f3..aed2bf9 100644
--- 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(
--
GitLab