场景适配方案
针对零件表面划痕、焊接缺陷等典型问题:
实施步骤:
- 数据准备阶段:
- 采集200+缺陷样本(需包含多种光照条件)
- 使用
labelme
标注缺陷区域(导出JSON格式)
- 模型选择:
- 高精度场景:选用
yoloe-v8l-seg.pt
预训练模型 - 实时产线:改用
yoloe-s-det.pt
纯检测模型
- 高精度场景:选用
- 部署优化:
- 通过
export.py
输出ONNX/TensorRT格式 - 设置
--conf 0.7
避免误检(质检需严格)
- 通过
典型工作流:
# 视觉提示模式(适合外观变异大的缺陷) python predict_visual_prompt.py --source conveyor.jpg --ref defect_sample1.jpg --checkpoint yoloe-l.pt
注意事项:对反光金属件,需先进行cv2.GaussianBlur
预处理
本答案来源于文章《YOLOE:实时视频检测和分割物体的开源工具》