KJNodes/InstanceDiffusiongenerated

DrawInstanceDiffusionTracking

该节点接收一个图像批次和一个 TRACKING 类型的跟踪数据作为输入

Draw Instance Diffusion Tracking

image
tracking
image
box_line_width
2
draw_text
font
COMBO
font_size
20
KJNodes

DrawInstanceDiffusionTracking 节点,该节点接收一个图像批次和一个 TRACKING 类型的跟踪数据作为输入。

DrawInstanceDiffusionTracking 节点用于在图像上可视化实例分割与跟踪的结果。其核心功能是读取由 CreateInstanceDiffusionTracking 节点生成的跟踪数据,并将这些数据(如物体的边界框和类别标签)绘制到对应的图像上,从而生成带有标注信息的可视化图像。

节点功能

该节点接收一个图像批次和一个 TRACKING 类型的跟踪数据作为输入。在内部,它使用 Matplotlib 的彩虹色谱为不同的跟踪类别分配颜色,并利用 PIL 库的绘图功能,将每个图像帧中检测到的物体边界框和可选的类别文本叠加到原始图像上。用户可以对绘制的样式,如框线宽度、字体和文本显示进行自定义。

节点参数说明 - DrawInstanceDiffusionTracking

连接输入 (Inputs)

参数名称数据类型必填默认值取值范围/选项说明
imageIMAGE--输入图像批次,用于在其上绘制跟踪数据。
trackingTRACKING--tracking(输入参数):来自 CreateInstanceDiffusionTracking 节点的跟踪数据,包含类别和边界框信息。
fontCOMBO--用于绘制文本的字体名称。从可用字体中选择。

控件参数 (Parameters)

参数名称数据类型必填默认值取值范围/选项说明
box_line_widthINT21 - 10 (步长: 1)box_line_width(输入参数):设置绘制边界框的线条宽度,取值范围为 1 到 10。
draw_textBOOLEAN-draw_text(输入参数):布尔开关,控制是否在边界框旁绘制类别标签文本。
font_sizeINT20-font_size(输入参数):设置绘制文本的字体大小。

输出

参数名称数据类型说明
imageIMAGEimage(输出参数):输出已绘制跟踪框和标签的可视化图像。

使用场景

在视频分析或序列图像处理的工作流中,此节点非常有用。例如,在处理完一段视频的所有帧并获得物体跟踪数据后,你可以使用此节点将跟踪框和 ID 信息逐帧绘制回原视频图像上,从而生成一个带有跟踪轨迹的可视化视频,便于直观地检查跟踪算法的准确性和稳定性。

注意事项

请注意,输入的 tracking 数据必须来自 CreateInstanceDiffusionTracking 节点,且其结构需要与输入的图像批次在时序上严格对应。如果跟踪数据中的帧索引超出了图像批次的范围,对应的绘制操作将被跳过。

DrawInstanceDiffusionTracking 节点源码链接

DrawInstanceDiffusionTracking 节点来自 ComfyUI-KJNodes 节点包。

评论

使用 GitHub 登录后即可参与讨论。

评论加载中…