DrawInstanceDiffusionTracking

Узел DrawInstanceDiffusionTracking принимает на вход пакет изображений и данные отслеживания типа TRACKING.

Узел DrawInstanceDiffusionTracking используется для визуализации результатов сегментации экземпляров и отслеживания на изображениях. Его основная функция — чтение данных отслеживания, сгенерированных узлом CreateInstanceDiffusionTracking, и нанесение этих данных (таких как ограничивающие рамки объектов и метки категорий) на соответствующие изображения, создавая таким образом визуализированные изображения с аннотациями.

Функциональность узла

Этот узел принимает на вход пакет изображений и данные отслеживания типа TRACKING. Внутренне он использует радужную цветовую карту Matplotlib для назначения цветов различным категориям отслеживания и использует функции рисования библиотеки PIL для наложения ограничивающих рамок обнаруженных объектов и, опционально, текста меток категорий на каждый кадр исходного изображения. Пользователь может настроить стиль рисования, например, ширину линии рамки, шрифт и отображение текста.

Описание параметров узла - DrawInstanceDiffusionTracking

Входные подключения (Inputs)

Название параметра	Тип данных	Обязательный	Значение по умолчанию	Диапазон значений/Опции	Описание
`image`	IMAGE	Да	-	-	Входной пакет изображений, на который наносятся данные отслеживания.
`tracking`	TRACKING	Да	-	-	tracking (входной параметр): Данные отслеживания от узла CreateInstanceDiffusionTracking, содержащие информацию о категориях и ограничивающих рамках.
`font`	COMBO	Да	-	-	Имя шрифта для рисования текста. Выберите из доступных шрифтов.

Параметры управления (Parameters)

Название параметра	Тип данных	Обязательный	Значение по умолчанию	Диапазон значений/Опции	Описание
`box_line_width`	INT	Да	2	1 - 10 (шаг: 1)	box_line_width (входной параметр): Устанавливает ширину линии для рисования ограничивающих рамок. Диапазон значений от 1 до 10.
`draw_text`	BOOLEAN	Да	True	-	draw_text (входной параметр): Логический переключатель, управляющий отображением текста метки категории рядом с ограничивающей рамкой.
`font_size`	INT	Да	20	-	font_size (входной параметр): Устанавливает размер шрифта для рисования текста.

Выходные данные (Output)

Название параметра	Тип данных	Описание
image	IMAGE	image (выходной параметр): Выводит визуализированное изображение с нанесенными рамками отслеживания и метками.

Сценарии использования

Этот узел очень полезен в рабочих процессах анализа видео или обработки последовательностей изображений. Например, после обработки всех кадров видео и получения данных отслеживания объектов можно использовать этот узел для нанесения рамок отслеживания и информации об ID на каждый кадр исходного видеоизображения, создавая таким образом визуализированное видео с траекториями отслеживания, что удобно для интуитивной проверки точности и стабильности алгоритма отслеживания.

Важные замечания

Обратите внимание, что входные данные tracking должны поступать от узла CreateInstanceDiffusionTracking, и их структура должна строго соответствовать входному пакету изображений по времени. Если индекс кадра в данных отслеживания выходит за пределы диапазона пакета изображений, соответствующая операция рисования будет пропущена.

Ссылка на исходный код узла DrawInstanceDiffusionTracking

Узел DrawInstanceDiffusionTracking из пакета узлов ComfyUI-KJNodes.