exception handle

src/diary_cover_similarity.py

@@ -58,20 +58,23 @@ def save_diary_image_info(save_file, face_to_vec_f):
     count = 0
     results = es_scan("diary", q)
     with open(save_file, "w") as f:
-        for item in results:
-            diary_id = item["_id"]
-            # before_cover_url = item["_source"]["before_cover_url"] + "-w"
-            after_cover_url = item["_source"]["after_cover_url"] + "-w"
-            print("count: " + str(count) + " " + str(diary_id))
-            img = url_to_ndarray(after_cover_url)
-            if img.any():
-                count += 1
+        try:
+            for item in results:
+                diary_id = item["_id"]
+                # before_cover_url = item["_source"]["before_cover_url"] + "-w"
+                after_cover_url = item["_source"]["after_cover_url"] + "-w"
                 print("count: " + str(count) + " " + str(diary_id))
-                faces = face_to_vec_f(img)
-                for face in faces:
-                    line = str(diary_id) + "\t" + face["feature"] + "\n"
-                    # print(line)
-                    f.write(line)
+                img = url_to_ndarray(after_cover_url)
+                if img.any():
+                    count += 1
+                    print("count: " + str(count) + " " + str(diary_id))
+                    faces = face_to_vec_f(img)
+                    for face in faces:
+                        line = str(diary_id) + "\t" + face["feature"] + "\n"
+                        # print(line)
+                        f.write(line)
+        except Exception as e:
+            print(e)
 
 
 def save_faiss_index(load_file, save_path):