qsitj commited on
Commit
15c2044
·
verified ·
1 Parent(s): fbb1f4d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -30,7 +30,7 @@ label_color_dict = {}
30
 
31
  def query_data(in_pil_img: Image.Image):
32
  results = detector(in_pil_img)
33
- print(f"检测结果:{results}")
34
  return results
35
 
36
 
@@ -105,7 +105,7 @@ def process_video(input_video_path):
105
  timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
106
  output_video_filename = f"output_{timestamp}.mp4"
107
  output_video_path = os.path.join(output_dir, output_video_filename)
108
- print(f"输出视频信息:{output_video_path}, {width}x{height}, {fps}fps")
109
  out = cv2.VideoWriter(output_video_path, fourcc, fps, (width, height))
110
 
111
  while True:
@@ -115,15 +115,15 @@ def process_video(input_video_path):
115
 
116
  rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
117
  pil_image = Image.fromarray(rgb_frame)
118
- print(f"Input frame of shape {rgb_frame.shape} and type {rgb_frame.dtype}") # 调试信息
119
  annotated_frame = get_annotated_image(pil_image)
120
  bgr_frame = cv2.cvtColor(annotated_frame, cv2.COLOR_RGB2BGR)
121
- print(f"Annotated frame of shape {bgr_frame.shape} and type {bgr_frame.dtype}") # 调试信息
122
  # 确保帧的尺寸与视频输出一致
123
  if bgr_frame.shape[:2] != (height, width):
124
  bgr_frame = cv2.resize(bgr_frame, (width, height))
125
 
126
- print(f"Writing frame of shape {bgr_frame.shape} and type {bgr_frame.dtype}") # 调试信息
127
  out.write(bgr_frame)
128
 
129
  cap.release()
 
30
 
31
  def query_data(in_pil_img: Image.Image):
32
  results = detector(in_pil_img)
33
+ # print(f"检测结果:{results}")
34
  return results
35
 
36
 
 
105
  timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
106
  output_video_filename = f"output_{timestamp}.mp4"
107
  output_video_path = os.path.join(output_dir, output_video_filename)
108
+ # print(f"输出视频信息:{output_video_path}, {width}x{height}, {fps}fps")
109
  out = cv2.VideoWriter(output_video_path, fourcc, fps, (width, height))
110
 
111
  while True:
 
115
 
116
  rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
117
  pil_image = Image.fromarray(rgb_frame)
118
+ # print(f"Input frame of shape {rgb_frame.shape} and type {rgb_frame.dtype}") # 调试信息
119
  annotated_frame = get_annotated_image(pil_image)
120
  bgr_frame = cv2.cvtColor(annotated_frame, cv2.COLOR_RGB2BGR)
121
+ # print(f"Annotated frame of shape {bgr_frame.shape} and type {bgr_frame.dtype}") # 调试信息
122
  # 确保帧的尺寸与视频输出一致
123
  if bgr_frame.shape[:2] != (height, width):
124
  bgr_frame = cv2.resize(bgr_frame, (width, height))
125
 
126
+ # print(f"Writing frame of shape {bgr_frame.shape} and type {bgr_frame.dtype}") # 调试信息
127
  out.write(bgr_frame)
128
 
129
  cap.release()