Spaces:
Running
on
Zero
Running
on
Zero
恢复预处理,之后再想想办法吧,傻逼HF
Browse files
app.py
CHANGED
@@ -12,10 +12,10 @@ model = YOLO('NailongKiller.yolo11n.pt').to(device)
|
|
12 |
|
13 |
@spaces.GPU
|
14 |
def predict(img):
|
15 |
-
#
|
16 |
-
|
17 |
# 规范化像素值到 0-1 范围
|
18 |
-
img_tensor = torch.tensor(
|
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)
|