Update api_server.py

api_server.py

@@ -75,45 +75,45 @@ def predict():
     if results is None or len(results) == 0:
         return jsonify({'error': 'No results from YOLO model'}), 400
 
-    # 渲染推理結果到圖像
-    img_with_boxes = results[0].plot()  # 使用 results[0]，假設只有一張圖像作推理
+    # # 渲染推理結果到圖像
+    # img_with_boxes = results[0].plot()  # 使用 results[0]，假設只有一張圖像作推理
 
-    # 將 numpy array 轉換為 PIL Image
-    img = Image.fromarray(img_with_boxes)
+    # # 將 numpy array 轉換為 PIL Image
+    # img = Image.fromarray(img_with_boxes)
 
-    # 儲存圖片到內存緩衝區
-    img_io = io.BytesIO()
-    img.save(img_io, 'PNG')
-    img_io.seek(0)
+    # # 儲存圖片到內存緩衝區
+    # img_io = io.BytesIO()
+    # img.save(img_io, 'PNG')
+    # img_io.seek(0)
 
-    # 返回處理後的圖像
-    return send_file(img_io, mimetype='image/png')
+    # # 返回處理後的圖像
+    # 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