场景适配方案
针对零件表面划痕、焊接缺陷等典型问题:
Etapas de implementação:
- 数据准备阶段:
- 采集200+缺陷样本(需包含多种光照条件)
- fazer uso de
labelme
标注缺陷区域(导出JSON格式)
- Seleção de modelos:
- 高精度场景:选用
yoloe-v8l-seg.pt
Modelo de pré-treinamento - 实时产线:改用
yoloe-s-det.pt
纯检测模型
- 高精度场景:选用
- 部署优化:
- aprovar (um projeto de lei ou inspeção etc.)
export.py
输出ONNX/TensorRT格式 - configurar
--conf 0.7
避免误检(质检需严格)
- aprovar (um projeto de lei ou inspeção etc.)
典型工作流:
# 视觉提示模式(适合外观变异大的缺陷) python predict_visual_prompt.py --source conveyor.jpg --ref defect_sample1.jpg --checkpoint yoloe-l.pt
Advertências:对反光金属件,需先进行cv2.GaussianBlur
pré-processamento
Essa resposta foi extraída do artigoYOLOE: uma ferramenta de código aberto para detecção de vídeo em tempo real e segmentação de objetosO