Draw Contour(s)

Documentation

• Class name: Draw Contour(s)
• Category: Bmad/CV/Contour
• Output node: False

The DrawContours node is designed for visualizing contours on images by drawing them over the original image. It allows for customization of the contour visualization, such as selecting specific contours to draw, adjusting the thickness of the contour lines, and choosing their color, thereby enhancing the interpretability of contour-based analyses.

Input types

Required

• image
  • The original image on which contours are to be drawn. It serves as the background for contour visualization.
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor
• contours
  • A collection of contours to be drawn on the image. Each contour is a sequence of points defining its shape.
  • Comfy dtype: CV_CONTOURS
  • Python dtype: List[List[Tuple[int, int]]]
• index_to_draw
  • Specifies which contour from the collection to draw. A value of -1 indicates that all contours should be drawn.
  • Comfy dtype: INT
  • Python dtype: int
• thickness
  • The thickness of the contour lines. A negative value indicates that contours should be filled.
  • Comfy dtype: INT
  • Python dtype: int
• color
  • The color of the contour lines. This allows for customization of the visual appearance of contours.
  • Comfy dtype: COLOR
  • Python dtype: Tuple[int, int, int]

Output types

• image
  • Comfy dtype: IMAGE
  • The image with the specified contours drawn over it, enhancing visual analysis of contours.
  • Python dtype: torch.Tensor

Usage tips

• Infra type: GPU
• Common nodes: unknown

Source code

class DrawContours:
    @classmethod
    def INPUT_TYPES(s):
        return {
            "required": {
                "image": ("IMAGE",),
                "contours": ("CV_CONTOURS",),
                "index_to_draw": ("INT", {
                    "default": -1,
                    "min": -1,
                    "max": 1000,
                    "step": 1
                }),
                "thickness": ("INT", {
                    "default": 5,
                    "min": -1,
                    "max": 32,
                    "step": 1
                }),
                "color": ("COLOR",),
            }
        }

    RETURN_TYPES = ("IMAGE",)
    FUNCTION = "draw"

    CATEGORY = "Bmad/CV/Contour"

    def draw(self, image, contours, index_to_draw, color, thickness):
        background = tensor2opencv(image)

        um_image = cv.UMat(background)
        cv.drawContours(um_image, contours, index_to_draw, ImageColor.getcolor(color, "RGB"), thickness)
        contour_image = um_image.get()

        image = opencv2tensor(contour_image)

        return (image,)