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

add self.

pull/119/head
KaiyangZhou 2019-01-28 17:29:22 +00:00
parent f045778602
commit daf184608c
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
torchreid/data_manager.py
View File

@ -69,6 +69,7 @@ class ImageDataManager(BaseDataManager):
self.num_instances = num_instances self.num_instances = num_instances
self.cuhk03_labeled = cuhk03_labeled self.cuhk03_labeled = cuhk03_labeled
self.cuhk03_classic_split = cuhk03_classic_split self.cuhk03_classic_split = cuhk03_classic_split
self.market1501_500k = market1501_500k
# Build train and test transform functions # Build train and test transform functions
transform_train = build_transforms(self.height, self.width, is_train=True) transform_train = build_transforms(self.height, self.width, is_train=True)
@ -82,7 +83,7 @@ class ImageDataManager(BaseDataManager):
for name in self.source_names: for name in self.source_names:
dataset = init_imgreid_dataset( dataset = init_imgreid_dataset(
root=self.root, name=name, split_id=self.split_id, cuhk03_labeled=self.cuhk03_labeled, root=self.root, name=name, split_id=self.split_id, cuhk03_labeled=self.cuhk03_labeled,
cuhk03_classic_split=self.cuhk03_classic_split, market1501_500k=market1501_500k cuhk03_classic_split=self.cuhk03_classic_split, market1501_500k=self.market1501_500k
) )
for img_path, pid, camid in dataset.train: for img_path, pid, camid in dataset.train:
@ -115,7 +116,7 @@ class ImageDataManager(BaseDataManager):
for name in self.target_names: for name in self.target_names:
dataset = init_imgreid_dataset( dataset = init_imgreid_dataset(
root=self.root, name=name, split_id=self.split_id, cuhk03_labeled=self.cuhk03_labeled, root=self.root, name=name, split_id=self.split_id, cuhk03_labeled=self.cuhk03_labeled,
cuhk03_classic_split=self.cuhk03_classic_split, market1501_500k=market1501_500k cuhk03_classic_split=self.cuhk03_classic_split, market1501_500k=self.market1501_500k
) )
self.testloader_dict[name]['query'] = DataLoader( self.testloader_dict[name]['query'] = DataLoader(