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