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

Merge pull request #85 from xieenze/master

Browse Source 
fix a bug of image_list.py
This commit is contained in:
Xiaohang Zhan 2020-12-25 16:16:27 +08:00 committed by GitHub
commit 1e1645ee9c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
openselfsup/datasets/data_sources/image_list.py
View File

@ -17,7 +17,7 @@ class ImageList(object):
self.fns, self.labels = zip(*[l.strip().split() for l in lines]) self.fns, self.labels = zip(*[l.strip().split() for l in lines])
self.labels = [int(l) for l in self.labels] self.labels = [int(l) for l in self.labels]
else: else:
assert self.return_label is False # assert self.return_label is False
self.fns = [l.strip() for l in lines] self.fns = [l.strip() for l in lines]
self.fns = [os.path.join(root, fn) for fn in self.fns] self.fns = [os.path.join(root, fn) for fn in self.fns]
self.memcached = memcached self.memcached = memcached