Ultralytics

🚀 ComfyUI Ultralytics統合 - Midnight1111モデルセット

このプロジェクトは主にUltralyticsモデルをComfyUIに統合し、ユーザーが物体検出などの操作を行いやすくすることを目的としています。簡単な手順でモデルの読み込みと使用が可能で、同時に遭遇する可能性のあるセキュリティ問題や読み込みエラーについても説明されています。

🚀 クイックスタート

モデルの読み込み手順

1. ComfyUIフォルダの特定：ComfyUIフォルダのパスを見つけます。例えば ~/ComfyUI/ または C:\ComfyUI\ です。
2. ディレクトリの作成：

   cd /path/to/ComfyUI/models
   mkdir -p ultralytics/segm ultralytics/bbox
   

3. .pt ファイルの配置：
   • セグメンテーションモデルファイル → models/ultralytics/segm/
   • 検出（バウンディングボックス）モデルファイル → models/ultralytics/bbox/
4. ComfyUIの再起動：

   cd /path/to/ComfyUI
   

5. 界面での操作： 界面で、ノードを追加 → モデル → Ultralytics → セグメンテーションまたはバウンディングボックスモデルを選択します。 画像ローダー → Ultralyticsノード → プレビューアを接続します。

📚 ドキュメント

不安全なファイルについて

getattr は危険なpickle関数として分類されているため、これを使用するすべてのセグメンテーションモデルは不安全と見なされます。ただし、すべてのモデルは公式のUltralyticsライブラリを使用して作成および保存されているため、信頼できるソースからダウンロードしたファイルは安全に使用できます。詳細情報については、https://huggingface.co/docs/hub/security-pickle を参照してください。

重みの読み込みに関する注意事項

以下のエラーが発生した場合：

Weights-only load failed. Unsupported global: builtins.set

これは、PyTorchが（セキュリティ上の理由から）デフォルトで特定のオブジェクトの読み込みをブロックしているためです。この問題を安全に解決するには、読み込み前に以下のコードを追加します：

import torch
torch.serialization.add_safe_globals([set])

これにより set がホワイトリストに追加され、weights_only=True を使用して安全に読み込むことができます。それ以外の場合は、ファイルを100%信頼する場合を除き、weights_only を無効にしないでください。詳細情報は PyTorchのtorch.loadドキュメント を参照してください。

📄 ライセンス

このプロジェクトはAGPL - 3.0ライセンスの下で提供されています。

📦 モデル情報