当前位置:AIGC资讯 > AIGC > 正文

ComfyUI中的sam模型国内下载方法

was-node-suite-comfyui这个节点安装的时候,有它内部的config配置文件,里面其实给了一些下载地址,配置文件里是这么写的:

"sam_model_vith_url": "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth",
"sam_model_vitl_url": "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_l_0b3195.pth",
"sam_model_vitb_url": "https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth",

实际上这些模型,hugging face上也有,例如:https://huggingface.co/spaces/abhishek/StableSAM/tree/main

但是如果网络不方便访问,国内直接找镜像,在hf-mirror.com网站上面,是无法访问space空间下载,所以上面的这种hg空间里的模型就无法直接下载了,找别的下载源就好。

魔搭社区里面正好有,可以很方便下载,地址如下:
https://modelscope.cn/models/yatengLG/ISAT_with_segment_anything_checkpoints/files

模型下载好以后,统一放在comfyui/models/sam下面。

总结

### 文章总结
本文围绕`was-node-suite-comfyui`节点安装过程中的模型下载问题进行了详细说明。文章指出,在安装此节点时,配置文件(`config`)内包含了指向特定模型文件的下载URL,这些模型文件是预训练好的Segment Anything模型(如SAM Vitamin系列的不同版本),用于计算机视觉任务中的细粒度分割。
#### 关键点概述:
1. **原生下载地址**:
- 配置文件中列出了每个SAM模型版本的官方下载URL,这些URL位于Facebook AI Research的公共文件服务器(`https://dl.fbaipublicfiles.com/segment_anything/`)。
2. **Hugging Face备选**:
- 尽管这些模型文件同样可在Hugging Face上找到(如`https://huggingface.co/spaces/abhishek/StableSAM/tree/main`),但当网络条件限制或需要国内访问时,直接从这些源下载可能存在困难。
3. **镜像源问题**:
- 针对Hugging Face的`hf-mirror.com`镜像网站在国内可能无法直接访问其Space空间进行下载,因此对于依赖这些空间的模型文件,需要寻找其他可靠的下载资源。
4. **魔搭社区资源**:
- 作为替代方案,文章推荐了魔搭社区(ModelScope,`https://modelscope.cn/`),该社区提供了`ISAT_with_segment_anything_checkpoints`等模型的下载链接。这些链接在国内访问稳定,且下载流程简便,是获取所需模型文件的优选途径。
5. **存储指导**:
- 下载模型后,文章强调需将模型文件统一放置在`comfyui/models/sam`路径下,以确保正确的加载和使用。
总结而言,本文为解决`was-node-suite-comfyui`节点安装中对Segment Anything模型的下载问题提供了多种备选方案,特别是对于国内用户,推荐使用魔搭社区的资源进行模型下载,以提高效率和便捷性。

更新时间 2024-08-21