mirrors
/
deep-person-reid
mirror of https://github.com/KaiyangZhou/deep-person-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update datamanager.py 

add **kwargs to all occurrences of `init_image_dataset ` in `datamanager.py`
pull/494/head
ZXYFrank 2022-02-23 21:23:47 +08:00 committed by GitHub
parent a6b2f77e8a
commit 94d6b02e40
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
torchreid/data/datamanager.py
View File

@ -248,7 +248,8 @@ class ImageDataManager(DataManager):
split_id=split_id, split_id=split_id,
cuhk03_labeled=cuhk03_labeled, cuhk03_labeled=cuhk03_labeled,
cuhk03_classic_split=cuhk03_classic_split, cuhk03_classic_split=cuhk03_classic_split,
market1501_500k=market1501_500k market1501_500k=market1501_500k,
**kwargs
) )
trainset_t.append(trainset_t_) trainset_t.append(trainset_t_)
trainset_t = sum(trainset_t) trainset_t = sum(trainset_t)
@ -297,7 +298,8 @@ class ImageDataManager(DataManager):
split_id=split_id, split_id=split_id,
cuhk03_labeled=cuhk03_labeled, cuhk03_labeled=cuhk03_labeled,
cuhk03_classic_split=cuhk03_classic_split, cuhk03_classic_split=cuhk03_classic_split,
market1501_500k=market1501_500k market1501_500k=market1501_500k,
**kwargs
) )
self.test_loader[name]['query'] = torch.utils.data.DataLoader( self.test_loader[name]['query'] = torch.utils.data.DataLoader(
queryset, queryset,
@ -319,7 +321,8 @@ class ImageDataManager(DataManager):
split_id=split_id, split_id=split_id,
cuhk03_labeled=cuhk03_labeled, cuhk03_labeled=cuhk03_labeled,
cuhk03_classic_split=cuhk03_classic_split, cuhk03_classic_split=cuhk03_classic_split,
market1501_500k=market1501_500k market1501_500k=market1501_500k,
**kwargs
) )
self.test_loader[name]['gallery'] = torch.utils.data.DataLoader( self.test_loader[name]['gallery'] = torch.utils.data.DataLoader(
galleryset, galleryset,