[Fix] Can not use 'img_info' without 'img_prefix' for data source. (o…

…pen-mmlab#491)
Betciso · Sep 30, 2022 · 58519b1 · 58519b1
1 parent ec0e563
commit 58519b1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmselfsup/datasets/data_sources/base.py b/mmselfsup/datasets/data_sources/base.py
@@ -95,8 +95,8 @@ def get_img(self, idx):
                 img_bytes,
                 flag=self.color_type,
                 channel_order=self.channel_order)
-        elif self.data_infos[idx].get('img_prefix', None) is not None:
-            if self.data_infos[idx]['img_prefix'] is not None:
+        elif self.data_infos[idx].get('img_info', None) is not None:
+            if self.data_infos[idx].get('img_prefix', None) is not None:
                 filename = osp.join(
                     self.data_infos[idx]['img_prefix'],
                     self.data_infos[idx]['img_info']['filename'])