Acesso no exterior: www.kdjingpai.com
Ctrl + D Marcar este site como favorito
Posição atual:fig. início " Respostas da IA

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

2025-08-27 1.4 K

自定义训练流程

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

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

  1. 准备10-20张带标注图片(COCO格式)
  2. modificaçõesclass_names.txt添加新类别(如”defective_part”)
  3. Em execução:
    python train_text_prompt.py --data custom.yaml --weights yoloe-s.pt

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

  • Passos:
    1. fazer uso degenerate_sam_masks.py生成参考图分割掩码
    2. realizartrain_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

Recomendado

Não consegue encontrar ferramentas de IA? Tente aqui!

Basta digitar a palavra-chave Acessibilidade Bing SearchA seção Ferramentas de IA deste site é uma maneira rápida e fácil de encontrar todas as ferramentas de IA deste site.

voltar ao topo

pt_BRPortuguês do Brasil