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許可證。

📦 模型信息