Nodes/ComfyUI YoloNasPose Tensorrt/Yolo Nas Pose Engine Builder
ComfyUI Node

Yolo Nas Pose Engine Builder

A ComfyUI node in tensorrt with 4 inputs and 1 output.

By yuvraj108c·Created 2 years ago·Updated about a year ago· 19
Yolo Nas Pose Engine Builder
    • message
    ◄model_sizelarge_0.5►
    ◄custom_engine_name►
    ◄use_fp16true►
    ◄custom_onnx_path►
    Categorytensorrt

    Inputs (4)

    NameTypeDefaultDescription
    model_sizeCOMBOlarge_0.5Select the YoloNasPose model size. Larger models provide better accuracy but require more VRAM.
    custom_engine_nameSTRINGOptional custom name for the TensorRT engine file. If empty, will use the default name based on the model.
    use_fp16BOOLEANtrueEnable FP16 precision for faster inference and lower VRAM usage. Disable if you experience stability issues.
    custom_onnx_pathSTRINGOptional path to a custom ONNX model file. If provided, will use this instead of downloading the predefined model.

    Outputs (1)

    NameTypeDescription
    messageSTRING—