Zugang aus Übersee: www.kdjingpai.com
Ctrl + D Lesezeichen für diese Seite
Derzeitige Position:Abb. Anfang " AI-Antworten

如何实现YOLOE对自定义物体的高精度检测?

2025-08-27 1.4 K

自定义训练流程

YOLOE支持通过三种方式适配新物体:

方案一:文本提示微调(5分钟速成)

  1. 准备10-20张带标注图片(COCO格式)
  2. Änderungenclass_names.txt添加新类别(如”defective_part”)
  3. Laufen:
    python train_text_prompt.py --data custom.yaml --weights yoloe-s.pt

方案二:视觉提示训练(适合无明确类别)

  • Schritte:
    1. ausnutzengenerate_sam_masks.py生成参考图分割掩码
    2. vollziehentrain_vp.py训练视觉编码器
    3. 预测时加载vp_model.pt作为参考

方案三:全模型训练(最高精度)

需准备>1000张标注图:

  1. 转换数据集:
    python tools/convert2yolo.py --data_path ./custom
  2. 生成缓存:
    python generate_grounding_cache.py --img-path ./train --json-path annotations.json
  3. 启动训练(推荐A100显卡):
    python train_seg.py --batch 64 --epochs 100

Empfohlen

Sie können keine AI-Tools finden? Versuchen Sie es hier!

Geben Sie einfach das Schlüsselwort Barrierefreiheit Bing-SucheDer Bereich KI-Tools auf dieser Website bietet eine schnelle und einfache Möglichkeit, alle KI-Tools auf dieser Website zu finden.

zurück zum Anfang

de_DEDeutsch