Zoe Depth Anything 预处理器类

最新推荐文章于 2025-03-08 17:57:41 发布

王马甲同学

最新推荐文章于 2025-03-08 17:57:41 发布

阅读量1.3k

点赞数 4

文章标签： python 人工智能

本文链接：https://blog.csdn.net/weixin_72569416/article/details/139881935

版权

报告文档：Zoe Depth Anything 检测器和预处理器类

导入语句

from ..utils import common_annotator_call, create_node_input_types
import comfy.model_management as model_management

这部分代码导入了通用的辅助函数以及模型管理模块，用于处理图像注释和模型的设备管理。

类 `ZoeDepthAnythingDetector` 的定义

构造函数

def __init__(self, model):
    self.model = model
    self.device = "cpu"

构造函数接收一个模型对象，并默认将设备设置为 “cpu”。

从预训练模型加载

@classmethod
def from_pretrained(cls, pretrained_model_or_path=DEPTH_ANYTHING_MODEL_NAME,
                    filename="depth_anything_metric_depth_indoor.pt"):
    model_path = custom_hf_download(pretrained_model_or_path, filename, subfolder="checkpoints_metric_depth",
                                    repo_type="space")
    conf = get_config("zoedepth", "infer")
    model = ZoeDepthAnything.buil