过滤0张或者1张图像

2021-11-12 15:25:43 +08:00 · 2021-11-12 15:25:43 +08:00 · 781b989407
parent fa20064a89
commit 781b989407
1 changed files with 3 additions and 2 deletions
--- a/projects/FastShoe/fastshoe/data/pair_dataset.py
+++ b/projects/FastShoe/fastshoe/data/pair_dataset.py
@ -37,6 +37,7 @@ class PairDataset(ImageDataset):
        pos_folders = []
        neg_folders = []
        for data in all_data:
+            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'])