diff --git a/research/cv/delf/src/dataset.py b/research/cv/delf/src/dataset.py
index 6d6e9f32251321f0a6d1e34055aa260a8fa76742..fa4527bf3d8f0026803eff86c243f48a7d279020 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 77b1bb2253db9ad182ce31ffafed880a3e33dbe4..bcc89a6a78d2cc07348b5309b044f7395b3c428e 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')