Hakureirm commited on
Commit
b482a00
·
verified ·
1 Parent(s): 27b4844

恢复预处理,之后再想想办法吧,傻逼HF

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -12,10 +12,10 @@ model = YOLO('NailongKiller.yolo11n.pt').to(device)
12
 
13
  @spaces.GPU
14
  def predict(img):
15
- # 直接使用输入图像,不进行resize
16
- img_array = np.array(img)
17
  # 规范化像素值到 0-1 范围
18
- img_tensor = torch.tensor(img_array, dtype=torch.float32).permute(2, 0, 1).unsqueeze(0).div(255.0).to(device)
19
 
20
  # 设置模型预测参数以加快后处理速度
21
  results = model.predict(
@@ -37,4 +37,4 @@ demo = gr.Interface(
37
  )
38
 
39
  if __name__ == "__main__":
40
- demo.launch(server_name="0.0.0.0", server_port=7860)
 
12
 
13
  @spaces.GPU
14
  def predict(img):
15
+ # 优化图像预处理
16
+ img_resized = np.array(Image.fromarray(img).resize((640, 640)))
17
  # 规范化像素值到 0-1 范围
18
+ img_tensor = torch.tensor(img_resized, dtype=torch.float32).permute(2, 0, 1).unsqueeze(0).div(255.0).to(device)
19
 
20
  # 设置模型预测参数以加快后处理速度
21
  results = model.predict(
 
37
  )
38
 
39
  if __name__ == "__main__":
40
+ demo.launch(server_name="0.0.0.0", server_port=7860)