Connected component labeling in SciPy

Introduction

Connected component labeling helps find groups of connected pixels in images or data. It tells which parts belong together.

Finding separate objects in a black and white image, like counting coins on a table.

Detecting clusters in a grid of data, such as patches of forest in satellite images.

Grouping connected regions in medical images, like tumors or organs.

Identifying connected areas in network graphs or maps.

Syntax

SciPy

from scipy.ndimage import label

labeled_array, num_features = label(input_array, structure=None)

input_array is a binary or boolean array where connected regions are labeled.

structure defines connectivity (e.g., 4-connectivity or 8-connectivity in 2D). If None, uses 8-connectivity in 2D.

Examples

Basic example with a small 2D array. Finds connected groups of 1s.

SciPy

import numpy as np
from scipy.ndimage import label

input_array = np.array([[1, 0, 0],
                        [1, 1, 0],
                        [0, 0, 1]])

labeled_array, num_features = label(input_array)
print(labeled_array)
print(num_features)

Edge case: empty array with no connected components.

SciPy

import numpy as np
from scipy.ndimage import label

input_array = np.zeros((3,3), dtype=int)

labeled_array, num_features = label(input_array)
print(labeled_array)
print(num_features)

Edge case: single element array with one connected component.

SciPy

import numpy as np
from scipy.ndimage import label

input_array = np.array([[1]])

labeled_array, num_features = label(input_array)
print(labeled_array)
print(num_features)

Using 8-connectivity to connect diagonally touching pixels.

SciPy

import numpy as np
from scipy.ndimage import label

input_array = np.array([[1, 1, 0],
                        [0, 1, 1],
                        [0, 0, 1]])

structure = np.array([[1,1,1],
                      [1,1,1],
                      [1,1,1]])  # 8-connectivity

labeled_array, num_features = label(input_array, structure=structure)
print(labeled_array)
print(num_features)

Sample Program

This program creates a 5x5 grid with three groups of connected 1s. It labels each group with a unique number and prints the results.

SciPy

import numpy as np
from scipy.ndimage import label

# Create a 5x5 binary array with three connected components
input_array = np.array([
    [1, 1, 0, 0, 0],
    [1, 1, 0, 1, 1],
    [0, 0, 0, 1, 1],
    [0, 1, 0, 0, 0],
    [0, 1, 1, 0, 0]
])

print("Input array:")
print(input_array)

# Label connected components with default connectivity
labeled_array, num_features = label(input_array)

print("\nLabeled array:")
print(labeled_array)
print(f"\nNumber of connected components: {num_features}")

Important Notes

Time complexity is roughly O(n), where n is the number of elements in the array.

Space complexity depends on the size of the input array and the labeled output array.

Common mistake: forgetting to use a binary array (only 0 and 1) as input.

Use connected component labeling when you want to identify and count distinct groups in data. For simple counting without location, other methods might be faster.

Summary

Connected component labeling finds groups of connected pixels in binary data.

It assigns a unique label to each connected group.

Useful for image analysis, clustering, and region detection.

Practice

1. What is the main purpose of connected component labeling in image processing?

easy

A. To enhance image contrast

B. To convert color images to grayscale

C. To identify and label groups of connected pixels in binary images

D. To compress image file size

Solution

1. Step 1: Understand connected component labeling

   It finds groups of connected pixels in binary images and assigns unique labels to each group.

2. Step 2: Compare with other image tasks

   Other options like grayscale conversion, contrast enhancement, and compression do not involve labeling connected pixels.

3. Final Answer:

   To identify and label groups of connected pixels in binary images -> Option C

4. Quick Check:

   Connected component labeling = identify connected pixel groups [OK]

Hint: Remember: labeling means assigning unique IDs to connected pixels [OK]

Common Mistakes:

• Confusing labeling with color conversion
• Thinking it compresses images
• Mixing it up with image enhancement

2. Which of the following is the correct way to import the connected component labeling function from scipy.ndimage?

easy

A. import scipy.ndimage.label

B. from scipy.ndimage import label

C. from scipy import label

D. import label from scipy.ndimage

Solution

1. Step 1: Recall correct import syntax in Python

   The correct syntax to import a function is 'from module import function'.

2. Step 2: Match with scipy.ndimage.label

   The function 'label' is inside 'scipy.ndimage', so 'from scipy.ndimage import label' is correct.

3. Final Answer:

   from scipy.ndimage import label -> Option B

4. Quick Check:

   Correct import syntax = from module import function [OK]

Hint: Use 'from module import function' to import specific functions [OK]

Common Mistakes:

• Using 'import scipy.ndimage.label' which is invalid
• Trying 'from scipy import label' which misses submodule
• Incorrect 'import label from scipy.ndimage' syntax

3. Given the following code, what will be the output of num_features?

import numpy as np
from scipy.ndimage import label

array = np.array([[1, 0, 0, 1],
                  [1, 1, 0, 0],
                  [0, 0, 1, 1],
                  [0, 0, 1, 0]])

labeled_array, num_features = label(array)
print(num_features)

medium

A. 1

B. 3

C. 4

D. 2

Solution

1. Step 1: Identify connected groups in the array

   There are two groups of connected 1s: one cluster at top-left (positions (0,0),(1,0),(1,1)) and one cluster at bottom-right (positions (0,3),(2,2),(2,3),(3,2)).

2. Step 2: Count the connected components

   Counting these groups gives 2 connected components.

3. Final Answer:

   2 -> Option D

4. Quick Check:

   Count connected 1s groups = 2 [OK]

Hint: Count distinct connected 1s clusters in the array [OK]

Common Mistakes:

• Counting isolated 1s as separate components incorrectly
• Ignoring connectivity rules
• Misreading array layout

4. What is wrong with the following code snippet for connected component labeling?

import numpy as np
from scipy.ndimage import label

binary_image = np.array([[1, 1, 0],
                         [0, 1, 0],
                         [1, 0, 1]])

labeled, num = label(binary_image, structure=1)
print(num)

medium

A. The 'structure' parameter should be an array, not an integer

B. The input array must be float type, not integer

C. The 'label' function does not accept a 'structure' parameter

D. The print statement syntax is incorrect

Solution

1. Step 1: Check 'structure' parameter type

   The 'structure' parameter expects an array defining connectivity, not a single integer.

2. Step 2: Identify correct usage

   Passing 'structure=1' is invalid; it should be an array like np.ones((3,3)) for full connectivity.

3. Final Answer:

   The 'structure' parameter should be an array, not an integer -> Option A

4. Quick Check:

   'structure' must be array, not int [OK]

Hint: Remember: 'structure' needs an array defining connectivity [OK]

Common Mistakes:

• Passing integer instead of array for 'structure'
• Assuming 'label' lacks 'structure' parameter
• Confusing data types of input array

5. You have a binary image with noise: small isolated pixels scattered randomly. How can you use connected component labeling with scipy to remove noise by keeping only components larger than 2 pixels?

hard

A. Label components, count sizes, then remove components with size ≤ 2

B. Apply Gaussian blur before labeling to remove noise

C. Use label function with parameter 'min_size=3' to filter components

D. Invert the image and label the background instead

Solution

1. Step 1: Label connected components in the binary image

   Use scipy.ndimage.label to assign unique labels to each connected group of pixels.

2. Step 2: Count the size of each labeled component and filter

   Calculate the size of each component and remove those with size less than or equal to 2 pixels to eliminate noise.

3. Final Answer:

   Label components, count sizes, then remove components with size ≤ 2 -> Option A

4. Quick Check:

   Filter small components after labeling to remove noise [OK]

Hint: Label, count sizes, remove small components to clean noise [OK]

Common Mistakes:

• Expecting label() to filter by size automatically
• Using image blur instead of component filtering
• Inverting image does not remove noise directly