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许可证。

📦 模型信息