mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

过滤0张或者1张图像

pull/608/head
zuchen.wang 2021-11-12 15:25:43 +08:00
parent fa20064a89
commit 781b989407
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
projects/FastShoe/fastshoe/data/pair_dataset.py
View File

@ -37,8 +37,9 @@ class PairDataset(ImageDataset):
pos_folders = []
neg_folders = []
for data in all_data:
pos_folders.append(data['positive_img_list'])
neg_folders.append(data['negative_img_list'])
if len(data['positive_img_list']) >= 2 and len(data['negative_img_list']) >= 1:
pos_folders.append(data['positive_img_list'])
neg_folders.append(data['negative_img_list'])
assert len(pos_folders) == len(neg_folders), self._logger.error('the len of self.pos_foders should be equal to self.pos_foders')
self.pos_folders = pos_folders