CavinHuang-comfyui-nodes-docs/docs/ONNXDetectorProvider.md

Documentation

• Class name: ONNXDetectorProvider
• Category: ImpactPack
• Output node: False
• Repo Ref: https://github.com/ltdrdata/ComfyUI-Impact-Pack.git

ONNXDetectorProvider节点旨在方便加载和提供用于检测任务的ONNX模型。它作为将基于ONNX的检测模型集成到工作流程中的接口，抽象了模型加载和设置的复杂性。

Input types

Required

• model_name
  • model_name参数对于识别要加载的特定ONNX模型至关重要。它确保使用了正确的模型进行检测任务，从而影响节点的执行和结果的准确性。
  • Comfy dtype: str
  • Python dtype: str

Output types

• BBOX_DETECTOR
  • BBOX_DETECTOR输出提供了一个配置好的ONNXDetector对象，用于执行对象检测任务。该对象封装了ONNX模型的功能，是检测过程中的核心组件。
  • Comfy dtype: ONNXDetector
  • Python dtype: ONNXDetector

Usage tips

• Infra type: CPU

Source code

class ONNXDetectorProvider:

    @classmethod
    def INPUT_TYPES(s):
        return {'required': {'model_name': (folder_paths.get_filename_list('onnx'),)}}
    RETURN_TYPES = ('BBOX_DETECTOR',)
    FUNCTION = 'load_onnx'
    CATEGORY = 'ImpactPack'

    def load_onnx(self, model_name):
        model = folder_paths.get_full_path('onnx', model_name)
        return (core.ONNXDetector(model),)