from ultralytics import YOLO
import os

# 设置工作目录
os.chdir("/home/bbit/mine/yolo/")

# 加载模型
model = YOLO("runs/detect/train13/weights/best.pt")

# 输入图片路径
img_path = "valid/raw_1.jpg"

# 推理
results = model(img_path)

# 生成输出路径（同目录下）
save_path = os.path.splitext(img_path)[0] + "_detected.jpg"

# 保存检测结果
results[0].save(filename=save_path)

print(f"检测结果已保存至: {save_path}")