mirrors/mmselfsup
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmselfsup.git synced 2025-06-03 14:59:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Fix]: Fix read image bug (#386)

Browse Source
This commit is contained in:
Yuan Liu 2022-07-29 19:23:18 +08:00 committed by GitHub
parent 3ac192dad2
commit abcd43b431
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mmselfsup/datasets/data_sources/base.py
View File

@ -109,10 +109,6 @@ class BaseDataSource(object, metaclass=ABCMeta):
channel_order=self.channel_order)
else:
img = self.data_infos[idx]['img']
img_bytes = self.file_client.get(filename)
img = mmcv.imfrombytes(
img_bytes, flag=self.color_type, channel_order=self.channel_order)
img = img.astype(np.uint8)
return Image.fromarray(img)