Update api_server.py

api_server.py

@@ -88,33 +88,33 @@ def predict():
 
     # 返回處理後的圖像
     return send_file(img_io, mimetype='image/png')
-  '''      
-    saved_images = []
+
+    # saved_images = []
     
-    # 儲存辨識後的圖片到指定資料夾
-    for result in results:
-        # 保存完整圖片
-        yolo_path = f'{YOLO_DIR}/results_{Path(result.path).name}'
-        result.save(yolo_path)
-        saved_images.append(yolo_path)
-
-        # 保存裁剪後的圖片（僅當 save_crop 返回有效的路徑時才加入）
-        cropped_images = result.save_crop(YOLO_DIR)  # 有些 YOLO 版本 save_crop 不返回值
-        if cropped_images:  # 確保不會對 None 進行迭代
-            if isinstance(cropped_images, list):  # 如果它返回一個列表
-                saved_images.extend(cropped_images)
-            else:
-                saved_images.append(cropped_images)
-
-    end_time = time.time()
-    inference_time = end_time - start_time
-
-    # 返回辨識結果的文件路徑以及推理時間
-    return jsonify({
-        'saved_images': saved_images,
-        'inference_time': inference_time
-    }), 200
-'''
+    # # 儲存辨識後的圖片到指定資料夾
+    # for result in results:
+    #     # 保存完整圖片
+    #     yolo_path = f'{YOLO_DIR}/results_{Path(result.path).name}'
+    #     result.save(yolo_path)
+    #     saved_images.append(yolo_path)
+
+    #     # 保存裁剪後的圖片（僅當 save_crop 返回有效的路徑時才加入）
+    #     cropped_images = result.save_crop(YOLO_DIR)  # 有些 YOLO 版本 save_crop 不返回值
+    #     if cropped_images:  # 確保不會對 None 進行迭代
+    #         if isinstance(cropped_images, list):  # 如果它返回一個列表
+    #             saved_images.extend(cropped_images)
+    #         else:
+    #             saved_images.append(cropped_images)
+
+    # end_time = time.time()
+    # inference_time = end_time - start_time
+
+    # # 返回辨識結果的文件路徑以及推理時間
+    # return jsonify({
+    #     'saved_images': saved_images,
+    #     'inference_time': inference_time
+    # }), 200
+
 
 
     # # dictionary is not a JSON: https://www.quora.com/What-is-the-difference-between-JSON-and-a-dictionary