yolo训练部署相关代码

ai/yolo/valid.py

@@ -0,0 +1,22 @@
+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}")