From c95f06a2dcbc3aa2b5edbcffdc7970165a34c822 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E7=B1=8D=E5=AE=B6=E8=8D=A3?= <jijiarong@huawei.com> Date: Tue, 11 Oct 2022 10:36:42 +0800 Subject: [PATCH] I5PS6P bug fix --- research/cv/delf/src/dataset.py | 7 ++++--- research/cv/delf/src/extract_feature.py | 5 +++-- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/research/cv/delf/src/dataset.py b/research/cv/delf/src/dataset.py index 6d6e9f322..fa4527bf3 100755 --- a/research/cv/delf/src/dataset.py +++ b/research/cv/delf/src/dataset.py @@ -38,9 +38,10 @@ def read_ground_truth(gt_file, image_file): index_list = [] num = 0 for name in names: - image_name = name.replace('.jpg', '') - index_list.append(image_name) - num += 1 + if name.endswith('.jpg'): + image_name = name.replace('.jpg', '') + index_list.append(image_name) + num += 1 path = gt_file names = os.listdir(path) diff --git a/research/cv/delf/src/extract_feature.py b/research/cv/delf/src/extract_feature.py index 77b1bb225..bcc89a6a7 100755 --- a/research/cv/delf/src/extract_feature.py +++ b/research/cv/delf/src/extract_feature.py @@ -248,8 +248,9 @@ def main(): names = os.listdir(args.images_path) image_paths = [] for name in names: - image_name = name.replace('.jpg', '') - image_paths.append(image_name) + if name.endswith('.jpg'): + image_name = name.replace('.jpg', '') + image_paths.append(image_name) num_images = len(image_paths) print(f'done! Found {num_images} images') -- GitLab