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

YOLOEでカスタムオブジェクトを高精度に検出するには?

2025-08-27 1.6 K

トレーニング・プロセスのカスタマイズ

YOLOEは、3つの方法で新しいオブジェクトの適応をサポートしている:

プログラムI:テキストプロンプトの微調整(5分間のクラッシュコース)

  1. ラベル付き画像を10~20枚用意する(COCOフォーマット)
  2. 修正class_names.txt新しいカテゴリーを追加する(例:"defect_part")
  3. 走っている:
    python train_text_prompt.py --data custom.yaml --weights yoloe-s.pt

プログラムII:視覚的手がかりトレーニング(不特定カテゴリー用)

  • ステップ
    1. 利用するgenerate_sam_masks.pyリファレンス・グラフ・セグメンテーション・マスクの生成
    2. はこびだすtrain_vp.py視覚エンコーダーのトレーニング
    3. 予測への負荷vp_model.pt参考

オプション3:フルモデルトレーニング(最高精度)

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ツールを素早く簡単に見つける方法です。

トップに戻る