Update app.py

app.py

@@ -45,32 +45,26 @@ fdic = {
     "weight" : "bold"
 }
 
+#######################################
 
-def infer(model, in_pil_img):
 
+def query_data(model, in_pil_img: Image.Image):
     results = None
     if model == "detr-resnet-101":
         results = detector101(in_pil_img)
     else:
         results = detector50(in_pil_img)
-
+    print(f"检测结果：{results}")
     return results
 
 
-#######################################
-
-
-def query_data(model, in_pil_img: Image.Image):
-    return infer(model, in_pil_img)
-
-
 
 def get_figure(in_pil_img):
     plt.figure(figsize=(16, 10))
     plt.imshow(in_pil_img)
     
     ax = plt.gca()
-
+    print(f"图像尺寸：{in_pil_img.size}")
     in_results = query_data(in_pil_img)
 
     for prediction in in_results:
@@ -88,18 +82,8 @@ def get_figure(in_pil_img):
     return plt.gcf()
 
 
-def infer(in_pil_img):
-    figure = get_figure(in_pil_img)
-
-    buf = io.BytesIO()
-    figure.savefig(buf, bbox_inches='tight')
-    buf.seek(0)
-    output_pil_img = Image.open(buf)
-
-    return output_pil_img
-
-
 def process_single_frame(frame):
+    print(f"开始处理单帧")
     # 将 BGR 转换为 RGB
     rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
     
@@ -118,6 +102,7 @@ def process_single_frame(frame):
 
 
 def infer_video(input_video_path):
+    print(f"开始处理视频 {input_video_path}")
     with tempfile.TemporaryDirectory() as tmp_dir:
         # output_video_path = Path(tmp_dir) / "output.mp4"
         cap = cv2.VideoCapture(input_video_path)