From 781b98940783d3318f52fe1dadfe6ac93fb935bf Mon Sep 17 00:00:00 2001 From: "zuchen.wang" Date: Fri, 12 Nov 2021 15:25:43 +0800 Subject: [PATCH] =?UTF-8?q?=E8=BF=87=E6=BB=A40=E5=BC=A0=E6=88=96=E8=80=851?= =?UTF-8?q?=E5=BC=A0=E5=9B=BE=E5=83=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- projects/FastShoe/fastshoe/data/pair_dataset.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/projects/FastShoe/fastshoe/data/pair_dataset.py b/projects/FastShoe/fastshoe/data/pair_dataset.py index 4bf3894..eee39ec 100644 --- a/projects/FastShoe/fastshoe/data/pair_dataset.py +++ b/projects/FastShoe/fastshoe/data/pair_dataset.py @@ -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