Computing connected components in an image

In this recipe, we will show an application of graph theory in image processing. We will compute connected components in an image. This method will allow us to label contiguous regions of an image, similar to the bucket fill tool of paint programs.

Finding connected components is also useful in many puzzle video games such as Minesweeper, bubble shooters, and others. In these games, contiguous sets of items with the same color need to be automatically detected.

How to do it…

  1. Let's import the packages:
    In [1]: import itertools
            import numpy as np
            import networkx as nx
            import matplotlib.colors as col
            import matplotlib.pyplot as plt
            %matplotlib inline
  2. We create a 10 x 10 image where each pixel can take one of ...

Get IPython Interactive Computing and Visualization Cookbook now with O’Reilly online learning.

O’Reilly members experience live online training, plus books, videos, and digital content from 200+ publishers.