海外からのアクセス:www.kdjingpai.com
Ctrl + D このサイトをブックマークする
現在の場所図頭 " AIアンサー

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

2025-08-27 1.4 K

自定义训练流程

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

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

  1. 准备10-20张带标注图片(COCO格式)
  2. 修正class_names.txt添加新类别(如”defective_part”)
  3. 走っている:
    python train_text_prompt.py --data custom.yaml --weights yoloe-s.pt

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

  • ステップ
    1. 利用するgenerate_sam_masks.py生成参考图分割掩码
    2. はこびだすtrain_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

おすすめ

AIツールが見つからない?こちらをお試しください!

キーワードを入力する アクセシビリティこのサイトのAIツールセクションは、このサイトにあるすべてのAIツールを素早く簡単に見つける方法です。

トップに戻る

ja日本語