Watershed (version 2.2.2, 2020-October-5)

Watershed.py
 
Version: 2.2.2
   
Author: Avinash Kak (kak@purdue.edu)
 
Date: 2020-October-5
 
 

Download Version 2.2.2:  gztar  

  		     Total number of downloads (all versions): 2404
     This count is automatically updated at every rotation of
     the weblogs (normally once every two to four days)
     Last updated: Tue Dec 10 06:02:01 EST 2024

View the main module code file in your browser  
View the ImageScanner code in your browser  
 
 
CHANGES:
  Version 2.2.2
 
    This version should not produce any warning messages when used with 
    Python 3.8.
 
  Version 2.2.1
 
    This version needed a couple of changes in order for the code to be
    consistent with the new style "getargs" format in the latest version of
    the Pillow library.
 
  Version 2.2.0
 
    This version of the module incorporates an ImageScanner class that
    facilitates the segmentation and detection of small objects in large
    images.  Consider the fact that you can now purchase a rather
    inexpensive digital camera that can record images of size 5472x3648.
    Let's say you use such a camera to take a photo of a large scene and
    that you are interested in segmenting out tiny objects in such
    photos. Creating solutions for such computer vision problems frequently
    requires scanning the large image and attempting object detection in
    the subimages for the different positions of the scanning window.  It
    is for such solutions to computer vision problems that you will find
    the ImageScanner class useful.  This class has been provided with
    methods that allow you to visualize the results of the operations
    carried out on the subimages both in the subimages and in the original
    large image simultaneously.
 
  Version 2.0.4
 
    This version fixes a "small bug" I discovered in the connected
    component labeling logic.  The bug made itself evident if a connected
    component consisted of just one pixel --- a highly unlikely occurrence
    in real-world images but important nevertheless.  Additionally, this
    version allows the 'apply_otsu_threshold_to_hue()' method to report the
    number of blobs found.  This method can be supplied with a keyword
    argument that serves as the area threshold for the blobs.
 
  Version 2.0.3
 
    This version allows you to carry out histogram equalization in an image
    before subjecting it to watershed segmentation.  Histogram equalization
    is applied to the 'v' component of the HSV representation of a color
    image.  Additionally, now you can first carry out a hue-based
    foreground/background separation through a threshold based on the Otsu
    algorithm and then use watershed to extract fine structure in just the
    foreground.
 
  Version 2.0.2
 
    This version enhances the module in the following two ways: (1) You can
    now apply watershed segmentation to just a portion of the input image.
    The portion that you are interested in can be specified through mouse
    clicks or by first clicking at a point and then dragging the mouse
    pointer over the image.  And (2) You can apply a color filter to the
    image before it is segmented.  The color filter can be specified in one
    of two different ways: as a triple of scalars or as a triple of
    pairs. If you specify the filter as a triple of three scalar values,
    each scalar is used to multiply the corresponding color component.  On
    the other hand, if you specify the filter as a triple of pairs, then a
    pixel is let through only if its each color component falls in the
    range dictated by the corresponding pair.
 
  Version 2.0.1
 
    With this version, you can now directly access the segmented blobs in
    your own code.  The previous versions only displayed the segmented
    image on your terminal screen.  This version also incorporates
    improvements in the implementation of the logic of watershed
    segmentation.
 
  Version 2.0
 
    This is a Python 3.x compliant version of the Watershed module.  This
    version should work with both Python 2.7 and Python 3.x.
 
  Version 1.1.2
 
    This version fixes the module packaging errors that had crept into the
    previous version.
 
  Version 1.1.1
 
    This version presents cleaned-up documentation.  The implementation
    code remains unchanged.
 
  Version 1.1
 
    This version fixes a bug in the dilate() and erode() methods of the
    module that caused these methods to misbehave for non-square images.
    Version 1.1 also includes improvements in the explanatory comments
    included in the scripts in the Examples directory.
 
 
INTRODUCTION:
 
    Over the years, the watershed algorithm has emerged as a powerful
    approach to image segmentation.  Image segmentation means to separate
    the foreground --- meaning the objects of interest --- from the
    background.  This approach becomes even more powerful when you allow a
    user to modify the image gradients prior to the application of the
    watershed algorithm.
 
    In the watershed algorithm, we think of the gradient image as a 3D
    topographic relief map that consists of mountains where the gradient
    values are high and valleys where they are low.  The watersheds are the
    ridge lines in this 3D map that delineate common basins for water
    drainage.  Note that the term watershed is also used to denote all of
    the surface area where the water drains toward a common basin (or
    valley).  When thought of as in the latter definition, watershed area
    would be delineated by the aforementioned ridge lines.  
 
    In computer vision algorithms, a watershed consists of the highest
    points that delineate a surface area in which the water would drain to
    a common valley.  It is important to realize that the points on a
    watershed can be at very different elevations. That is, the height
    above the ground plane along a watershed can vary significantly. Think
    of the Great Continental Divide as a watershed that divides the two
    American continents, the North and the South, into two regions, one
    that drains into the Pacific and the other that drains into the
    Atlantic.  If you were to walk along this Great Divide all the way from
    where it begins in Alaska to its final point in Patagonia at the
    southern tip of Chile, you will be at vastly different elevations above
    the sea level.  If you are on the Divide in Colorado, you are likely to
    find snow at several places even in summer months.  But if you follow
    the divide into Arizona, at several places you'll be in the middle of
    what to the naked eye would like a flat desert.
 
    These properties of the natural watersheds make the Watershed a
    powerful paradigm for image segmentation.  When you closely examine the
    semantically meaningful regions in typical images, you are likely to
    see a large variation in the gradient levels that separate the
    foreground regions from the background regions.  Obviously, the
    strength of the gradient that separates the foreground and the
    background is a function of the local gray levels in both the
    foreground and the background in the vicinity of the border between
    them.  So, while you cannot attribute specific values to such
    separating gradients, the gradient highlights are nonetheless real for
    the most part since that's how the human eye sees the different regions
    in an image.
 
    That brings us to the question of how to actually implement the
    watershed paradigm for image segmentation.  This module is based on
    what is known as the "flooding model" of computations for extracting
    the watershed from a gradient image when it is thought of as a 3D
    relief.  The flooding model is based on the idea that if we prick holes
    at the lowest points in all of the valleys of a topographic relief
    representation of a gradient image and gradually submerge the holed 3D
    structure in a pool of water, the rising flood in one valley will meet
    the flood in another valley at the watershed ridge points between the
    two valleys.  Since the watershed ridges can be at varying heights, in
    order to identify the watershed points beyond the first such point, we
    immediately put up a dam at the discovered watershed points so that the
    logic of watershed identification can be maintained at all heights.
 
    The flooding model that is implemented in this module is based on the
    work of Beucher, Lantuejoul, and Meyer.  A description of the algorithm
    can be found in the following Google accessible report by S. Beucher:
    "The Watershed Transformation Applied to Image Segmentation."  This
    algorithm will be refered to as the "BLM algorithm" in the rest of this
    web page.
 
    The BLM algorithm uses morphological operators for simulating the
    rising flood in the topographic relief map of the gradient image. The
    most important morphological operations in this context are those of
    dilations, distance mapping, calculation of the influence zones in a
    binary blob with respect to marker sets, and the determination of
    geodesic skeletons.  These operators are applied to the Z level sets
    derived from the gradient map.  For a given level, the pixels that
    belong to the Z set at that level are those whose gradient values are
    less than or equal to the level.  Flooding in the BLM algorithm is
    simulated by computing the influence zones of the flooded pixels at one
    Z level in all the pixels that belong to the next higher Z level.
 
 
COMPARISON WITH THE OpenCV WATERSHED IMPLEMENTATION:
 
    If your main interest is in just the final output --- good-quality
    watershed segmentations based on user-supplied seeds --- then this
    Python module is not for you and you should use the OpenCV
    implementation. With regard to the speed of execution, a pure Python
    module such as this cannot hope to compete with the C-based
    implementation in the OpenCV library.
 
    But do bear in mind that the segmentation produced by the OpenCV
    implementation is driven entirely by the user-supplied seeds.  In fact,
    the number of image segments produced by the OpenCV algorithm equals
    the number of seeds supplied by the user --- even when two different
    seeds are placed in the same homogeneous region of the image.
 
    On the other hand, this Python module will give you a watershed
    segmentation even when you do not supply any seeds (or, marks, as I
    refer to them in the implementation here).  So you could say that the
    user supplied marks (seeds) for this Python module are more for the
    purpose of creating new valleys in the topographic relief
    representation of the gradient map than for nucleating the formation of
    valleys for the discovery of watersheds that must partition the image
    into a specific number of segments.
   
 
INSTALLATION:
 
    The Watershed class was packaged using setuptools.  For installation,
    execute the following command in the source directory (this is the
    directory that contains the setup.py file after you have downloaded and
    uncompressed the package):
 
            sudo python setup.py install
 
    and/or, for the case of Python3, 
 
            sudo python3 setup.py install
 
    On Linux distributions, this will install the module file at a location
    that looks like
 
             /usr/local/lib/python2.7/dist-packages/
 
    and, for the case of Python3, at a location that looks like
 
             /usr/local/lib/python3.5/dist-packages/
 
    If you do not have root access, you have the option of working directly
    off the directory in which you downloaded the software by simply
    placing the following statements at the top of your scripts that use
    the Watershed class:
 
            import sys
            sys.path.append( "pathname_to_Watershed_directory" )
 
    To uninstall the module, simply delete the source directory, locate
    where the Watershed module was installed with "locate Watershed" and
    delete those files.  As mentioned above, the full pathname to the
    installed version is likely to look like
    /usr/local/lib/python2.7/dist-packages/Watershed*
 
    If you want to carry out a non-standard install of the Watershed
    module, look up the on-line information on Disutils by pointing your
    browser to
 
              http://docs.python.org/dist/dist.html
 
 
USAGE:
 
    To segment an image, you would first construct an instance of the
    Watershed class and invoke the methods shown below on this instance:
 
        from Watershed import *
 
        shed = Watershed(
                   data_image = "orchid0001.jpg",
                   binary_or_gray_or_color = "color",
                   size_for_calculations = 128,
                   sigma = 1,
                   gradient_threshold_as_fraction = 0.1,
                   level_decimation_factor = 16,
                   padding = 20,
               )
        shed.extract_data_pixels()
        shed.display_data_image()
        shed.mark_image_regions_for_gradient_mods()                     #(A)
        shed.compute_gradient_image()
        shed.modify_gradients_with_marker_minima()                      #(B)
        shed.compute_Z_level_sets_for_gradient_image()
        shed.propagate_influence_zones_from_bottom_to_top_of_Z_levels()
        shed.display_watershed()
        shed.display_watershed_in_color()
        shed.extract_watershed_contours()
        shed.extract_watershed_contours_with_random_sampling(30, 50)    #(C)
        shed.display_watershed_contours_in_color()
 
    
    The statements in lines (A) and (B) are needed only for marker-assisted
    segmentation with the module.  For a fully automated implementation of
    the BLM algorithm, you would need to delete those two statements.
 
    The statement in line (C) can be replaced by
 
        shed.extract_watershed_contours_separated()
 
    for faster contour extraction.  However, if the different contours are
    expected to share pixels, you are likely to get superior results with
    the call shown in line (C) above.  Both contour extraction methods
    return a list of contours, with each contour a list of pixel
    coordinates.
 
    If you just want to use this module just for demonstrating basic
    morphological operations of dilations and erosions, your script would
    look like:
 
        shed = Watershed(
                   data_image = "triangle1.jpg", 
                   binary_or_gray_or_color = "binary",
               )
        shed.extract_data_pixels() 
        shed.display_data_image()
        dilated_image = shed.dilate(5)                                     
        shed.erode(dilated_image, 5)        
 
    On the other hand, if you want to use this module to demonstrate
    distance mapping of a binary blob with respect to one or more marker
    blobs, your code is likely to look like:
 
        shed = Watershed(
                   data_image = "artpic3.jpg",
                   binary_or_gray_or_color = "binary",
                   debug = 0,
               )
        shed.extract_data_pixels() 
        shed.display_data_image()
        shed.connected_components("data")
        shed.mark_blobs()
        shed.connected_components("marks")
        shed.dilate_mark_in_its_blob(1)
 
    Note that now you must make calls to "connected_components()" to
    separate out the blobs in your input binary image, and to
    "mark_blobs()" to let a user mark up a blob for distance mapping.  It
    is the last call shown above, to "dilate_mark_in_its_blob()" that
    carries out distance mapping of the chosen blob with respect to the
    mark. Note that this takes an integer argument that is supposed to be
    integer index of the mark.  If you placed only one mark in the blob,
    this arg must be set to 1.  On the other hand, if you placed, say, two
    marks in a blob, then by supplying 2 for the argument you will see
    distance mapping with respect to the other mark. So, what you supply
    for the argument is an integer value between 1 and the number of marks
    you created.
 
    If you want to demonstrate the calculation of influence zones with this
    module, replace the last statement in the example shown above with the
    statement:
 
        shed.compute_influence_zones_for_marks()
 
    The module also includes two static methods that allow you to create
    your own binary images for demonstrating the basic operations built
    into the module.  The syntax for calling these method looks like:
 
        Watershed.gendata("triangle", (100,100), (10,10), 30, "new_triangle.jpg" 
 
        Watershed.make_binary_pic_art_nouveau("new_art_from_me")
 
    The second call in particular allows you to create artsy looking binary
    blobs just by rapidly moving your mouse over the window that is shown
    as you keep your left mouse button pressed.  Alternating clicks of the
    left mouse button start and stop this process.
 
 
CONSTRUCTOR PARAMETERS: 
 
    binary_or_gray_or_color: Must be set to either 'binary', or 'gray', or
                    'color', as the case may be.  A binary image is
                    thresholded after it is loaded in to produce binary
                    pixels.  And a color image is converted into a
                    grayscale image before the application of the watershed
                    algorithm.  Ordinarily, you would want to use binary
                    images just for demonstrating the dilation, erosion,
                    distance mapping, IZ calculation, and geodesic skeleton
                    calculation capabilities of this module.
 
    color_filter: The color filter can be specified in one of two different
                    ways: as a triple of scalars or as a triple of
                    pairs. If you specify the filter as a triple of three
                    scalar values, each scalar is used to multiply the
                    corresponding color component.  On the other hand, if
                    you specify the filter as a triple of pairs, then a
                    pixel is let through only if its each color component
                    falls in the range specified by the corresponding pair.
 
    data_image: The image you wish to segment as, say, a '.jpg' file
 
    gradient_threshold_as_fraction: This parameter when set allows the
                    system to ignore small gradients in the image.  Note
                    that the gradient values are normalized to be between 0
                    and 255, both ends inclusive.  The default for this
                    parameter is 0.1.
 
    level_decimation_factor: This factor controls the number of levels of
                    the gradient that are subject to watershed
                    calculations.  Recall that the image gradient is
                    normalized to values between 0 and 255.  If you set
                    level_decimation_factor to 8, only every 8th gradient
                    level will be considered for the flooding calculations.
                    That is, with level_decimation_factor set to 8, you
                    will have a total of 32 levels of the gradient for the
                    discovery of the watersheds.
 
    max_gradient_to_be_reached_as_fraction: This parameter is useful for
                    debugging purposes.  When set to, say, 0.5, it will
                    stop the rising flood at half of the maximum value for
                    image gradient.  So by setting this parameter to a
                    small value, you can carry out a more detailed
                    examination of the propagation of the influence zones
                    from one level to the next.
 
    padding: This parameter pads the input image with a band of zeros, the
                    width of the band being equal to the value of this
                    parameter.  You will get better results for the
                    watershed contours if you use this parameter.  The band
                    needs to be wide enough so that even after LoG
                    smoothing, the border pixels for the padded image are
                    still set to to zero.
 
    size_for_calculations: If the larger of the two image dimensions is
                    greater than this number, the image is reduced in size
                    so that its larger dimension corresponds to the number
                    supplied through this parameter. As you would expect,
                    the smaller this parameter, the faster your results.
                    The default for this parameter is 128.
 
    sigma: Controls the size of the Gaussian kernel used for smoothing the
                    image before its gradient is calculated.  Assuming the
                    pixel sampling interval to be unity, a sigma of 1 gives
                    you a 7x7 smoothing operator with Gaussian weighting.
                    The default for this parameter is 1.
 
 
PUBLIC METHODS:
 
    (1)  apply_color_filter_hsv()
 
         This method applies a color filter to the HSV representation of a
         color image prior to its watershed segmentation.  The filter must
         be specified either as a triple of scalars, as in "(1,0.5,0.8)",
         or a triple of pairs, as in "((0,35),(0,255),(0,255))".  When
         specified as a triple of scalars, each color component is
         multiplied by the corresponding scalar.  And when specified as a
         triple of pairs, each color component must be within the range
         dictated by the corresponding pair. With HSV, you are more likely
         to use the second form of the filter.  For example, the filter
         "((0,35),(0,255),(0,255))" works well if you want to let through
         only the red and reddish pixels.
 
 
    (2)  apply_color_filter_rgb()
 
         This method applies a color filter to the RGB representation of a
         color image prior to its watershed segmentation.  The filter must
         be specified either as a triple of scalars, as in "(1,0.5,0.8)",
         or a triple of pairs, as in "((0,35),(0,255),(0,255))".  When
         specified as a triple of scalars, each color component is
         multiplied by the corresponding scalar.  And when specified as a
         triple of pairs, each color component must be within the range
         dictated by the corresponding pair. With RGB, you are more likely
         to use the first form of the filter.  For example, if you wanted
         to apply the watershed segmentation to just the R component of a
         color image, your filter would be "(1,0,0)".
 
 
    (3)  apply_otsu_threshold_to_hue( polarity=1, area_threshold=0 )
 
         Applies the optimum Otsu threshold to hue in order to separate the
         image into foreground and background.  If the value supplied
         through the parameter 'polarity' is -1, all pixels whose hue is
         below the optimum Otsu threshold are accepted.  On the other hand,
         if the polarity is set to '1', all pixels whose hue is higher than
         the Otsu threshold are accepted. The method also isolates out the
         blobs resulting from the application of the Otsu threshold.  All
         blobs whose areas fall below the value supplied for
         'area_threshold' are rejected.  The method returns the number of
         blobs found.
 
 
    (4)  compute_influence_zones_for_marks()
 
         Calculates the influence zones in a binary blob with respect to
         the marks placed inside the blob.  The method also identifies the
         pixels at the geodesic skeleton formed by the influence zones.
 
 
    (5)  compute_LoG_image()
 
         This method computes the Laplacian-of-Gaussian (LoG) of an image.
         The LoG image is calculated as the difference of two
         Gaussian-smoothed versions of the input image at two slightly
         difference scales.  The LoG itself is NOT used in watershed
         calculations.
 
 
    (6)  compute_Z_level_sets_for_gradient_image()
 
         This method computes the Z levels for the BLM watershed algorithm.
         A pixel belongs to a specified Z level if the value of the image
         gradient at the pixel is less than or equal to that level.
 
 
    (7)  connected_components(arg)
 
         where 'arg' is the string "data" if you want to carry out
         connected-components labeling of binary blobs.  When the same
         method is called for the binary marks created by mouse clicks, we
         change the value of 'arg' to "marks".  Obviously, the components
         labeling logic works exactly the same in both cases.  The only
         differences are in how the labels are saved for bookkeeping
         purposes.
 
 
    (8)  dilate(radius, shape)
 
         where 'radius' would generally be a small integer denoting the
         radius of a disk structuring element to be used for the purpose of
         dilating the input pattern.  The parameter 'shape' must be set to
         either "square" or "circular" to specify the shape of the
         structuring element.  NOTE: This method only makes sense for
         binary input images, since it only carries out binary dilations.
 
 
    (9)  dilate_mark_in_its_blob()
 
         This is the method that demonstrations the distance mapping
         notions used in this module.  The module carries out a distance
         transformation of the blob that the user clicked on and does so
         with respect to the mark placed in the chosen blob when the left
         button of the mouse was clicked in it.
 
 
    (10) display_all_segmented_blobs()
 
         This method creates a composite image consisting of rectangular
         cells.  Each segmented object is placed in a separate cell.
 
 
    (11) display_data_image():
 
         It is always good to call this method after you have invoked
         "extract_data_pixels()" just to confirm the output of the data
         extraction step.
 
 
    (12) display_watershed()
 
         This method displays the watershed pixels against the image that
         was actually used for the application of the watershed algorithm.
         Even when the input image is in color, the image used for
         calculations is a grayscale version of the input
         image. Additionally, depending on the constructor parameter
         'size_for_calculations', the image used for calculations may also
         be of reduced size.
 
 
    (13) display_watershed_contours_in_color()
 
         The boundary contours for the segmented regions of the image are
         displayed against the original image by this method.
 
         To write this result image as a jpeg file to the disk, click on
         the "save" and "exit" buttons at the bottom of the window --- as
         opposed to just closing the window.  You can subsequently display
         the image from the disk file by using, say, the 'display' function
         from the ImageMagick library.  You can create a hardcopy version
         of the result by sending the jpeg image to a printer.
 
 
    (14) display_watershed_in_color()
 
         This display method shows the watershed pixels in color against
         the original image supplied to the module for segmentation.
 
         To write this result image as a jpeg file to the disk, click on
         the "save" and "exit" buttons at the bottom of the window --- as
         opposed to just closing the window.  You can subsequently display
         the image from the disk file by using, say, the 'display' function
         from the ImageMagick library.  You can create a hardcopy version
         of the result by sending the jpeg image to a printer.
 
 
    (15) erode(radius, shape)
 
         where 'radius' would generally be a small integer denoting the
         radius of a disk structuring element to be used for the purpose of
         dilating the input pattern.  The parameter 'shape' must be set to
         either "square" or "circular" to specify the shape of the
         structuring element.  NOTE: This method only makes sense for
         binary input images, since it only carries out binary dilations.
 
 
    (16) histogram_equalize():
 
        Carries out contrast enhancement in an image.  If an image is
        generally too dark or generally too bright, you may wish to
        histogram equalize it before subjecting it to watershed
        segmentation.
 
 
    (17) extract_data_pixels():
 
         This is the very first method you should call in all cases.  This
         loads your image into the module. If you declared your input image
         to be binary (when your image is in, say, the Jpeg format, the
         actual pixel values will in general not be binary), the loaded
         image is binarized by thresholding it in the middle of the gray
         scale range.  If you declared your input image to be color, it is
         first converted into grayscale for the calculation of the
         watersheds.  The image may also be reduced in size if that step is
         dictated by the value you supplied for the constructor parameter
         'size_for_calculations'.  Depending on this parameter, an input
         image declared to be gray will also be downsized.
 
 
    (18) extract_image_region_interactively_through_mouse_clicks():
 
         This method allows a user to use a sequence of mouse clicks in
         order to specify a region of the input image that should be
         subject to watershed segmentation.  The mouse clicks taken
         together define a polygon. The method encloses the polygonal
         region by a minimum bounding rectangle, which then becomes the new
         input image for the rest of processing.
 
 
    (19) extract_image_region_interactively_by_dragging_mouse():
 
         This is one method you can use to apply watershed segmentation to
         just a portion of your image.  This method extract the portion you
         want.  You click at the upper left corner of the rectangular
         portion of the image you are interested in and you then drag the
         mouse pointer to the lower right corner.  Make sure that you click
         on "save" and "exit" after you have delineated the area.
 
 
    (20) extract_rectangular_masked_segment_of_image(self, horiz_start, horiz_end, vert_start, vert_end)
 
         This method is useful for extracting a rectangular section from
         the data image supplied to the module.  Now how the parameters
         are laid out.  The first two parameters together specify the 
         width of the rectangle and the last two parameters together 
         the height.
 
 
    (21) extract_segmented_blobs_using_contours( min_length )
 
         This method uses the inside/outside logic at a point (based on the
         number of times a line through the point intersects the contour)
         in order to decide whether the point is inside a closed contour or
         outside it.  A contour must be longer than 'min_length' pixels in
         order to be considered for this logic. This method returns a list
         of lists, with each list in the latter being a list of all the
         pixels that are in a segmented blob.
 
 
    (22) extract_segmented_blobs_using_region_growing( num_of_segments, min_area )
 
         This method uses region growing to pull out the segmented blobs
         from the output of watershed segmentation.  All such segments
         whose area (in terms of the number of pixels) is less than the
         parameter 'min_area' are discarded.  Set the first parameter,
         'num_of_segments', to the number of segmented blobs you wish to
         extract from the image.  The value given to 'num_of_segments' sets
         a maximum for the number of segmented blobs the method will seek
         out from the watershed segmentation.  This method returns a list
         of lists, with each list in the latter being a list of all the
         pixels that are in a segmented blob.
 
 
    (23) extract_watershed_contours_with_random_sampling( how_many, min_len )
 
         This method extracts the watershed pixels as boundary contours
         using an 8-connected boundary following algorithm. Use this method
         if the different contours are expected to share pixels.  The
         parameter 'how_many' is the number of contours you'd like to
         extract and the parameter 'min_len' sets the smallest acceptable
         length for the contours.  You can use 'min_len' to reject small
         noise induced contours.  This method returns a list of contours,
         with each contour a list of pixel coordinates.
 
 
    (24) extract_watershed_contours_separated()
 
         This much faster method extracts the watershed pixels as boundary
         contours using an 8-connected boundary following algorithm. Use
         this method if all contours are disjoint.  Like the previous
         method, this method also returns a list of contours, with each
         contour a list of pixel coordinates.
 
 
    (25) gendata(feature, size_tuple, location_tuple, orientation_or_radius, filename)
 
         This static method is useful for generating simple binary patterns
         for checking the validity of the logic used for dilation, erosion,
         IZ calculation, geodesic skeleton calculation, etc.  The
         permissible values for the 'feature' parameter are: 'line',
         'triangle', 'rectangle', 'broken_rectangle', 'disk' and
         'twin_rect'.  The parameter 'size_tuple' is for the size of the
         output image desired.  The parameter 'position' is supposed to be
         a tuple (x,y) of pixel coordinates for specifying the position of
         the binary pattern in your image.  The parameter
         'orientation_or_radius' is an integer value specifying the number
         of degrees of rotation that should be applied to the pattern for a
         given 'feature' for the case of 'line', 'triangle', 'rectangle'
         and 'broken_rectangle'.  The same parameter for the case of
         'circle' means the radius of the disk. Finally, the parameter
         'filename' names the file in which the binary image will be
         deposited.  For the parameters that take tuple values, the first
         coordinate is along the horizontal direction pointing to the right
         and the second coordinate is along vertical direction pointing
         downwards.  The origin is at the upper left corner of your
         terminal screen.
 
 
    (26) make_binary_pic_art_nouveau( filename )
 
         This static method can be used to make "fun" binary images for
         demonstrating distance mapping of binary blobs, calculation of
         influence zones, etc.  This method is taken from Chapter 13 of my
         book "Scripting with Objects".
 
 
    (27) mark_blobs( purpose )
 
        A useful method for demonstrating distance mapping of a binary blob
        with respect to a marker blob, this method allows a user to both
        select one or more blobs in a binary image for the purpose of
        distance mapping and to also place marks on the blobs.  The
        'purpose' parameter must take one of the following two values:
        'distance_mapping' or 'influence_zones'.
 
 
    (28) mark_blobs_no_image_scale_change()
 
        The previously listed 'mark_blobs()' method adjusts the scale of
        the image according to your screen size to make it more convenient
        to enter the marks.  However, if you do not interact with the image
        without its scale being changed, use this method instead.
 
 
    (29) mark_image_regions_for_gradient_mods()
 
         For mark-based segmentation with the BLM watershed algorithm, this
         method elicits mouse clicks from the user that demarcate polygonal
         regions in the image where the gradient should be modified prior
         to the flooding process.  The mouse clicks must be supplied in a
         clockwise fashion to demarcate the polygonal regions.
 
 
    (30) modify_gradients_with_marker_minima()
 
         For mark-based application of the Watershed algorithm, it is this
         method that actually modifies the gradient image after a user has
         defined the regions for that purpose through the mouse clicks
         elicited by the mark_image_regions_for_gradient_mods() method.  In
         the current module, this modification simply consists of setting
         the gradient values to zero in such regions.
 
 
    (31) propagate_influence_zones_from_bottom_to_top_of_Z_levels()
           
         This is the workhorse of the module for watershed based
         segmentation of an image.  As explained elsewhere in this
         documentation, this method starts at the lowest Z level to
         discover the lowest valleys in the topographic relief
         representation of the image gradients.  Subsequently, the notion
         of a rising flood is simulated by calculating the influence zones
         (IZ) of the flooded pixels for one Z level in all of the pixels
         that belong to the next Z level.  The geodesic skeletons formed by
         the IZs lead to the discovery of the watershed pixels in a
         gradient image.
 
 
APPLYING SEGMENTATION TO AN IMAGE PORTION AND USING A COLOR FILTER:
 
    Starting with Version 2.0.2, you can apply the segmentation algorithm
    to just a portion of the input image.  You can specify the image
    portion through two different ways: (1) You can create a rectangular
    portion by clicking your mouse pointer at the upper left corner of the
    rectangular area you are interested in and then dragging the mouse
    pointer to its lower right corner. And (2) You can specify any
    polygonal shape by simply clicking clockwise at the points
    corresponding to the vertices of the polygon.  The method to call for
    the first way of defining an area for segmentation is:
    
        extract_image_region_interactively_by_dragging_mouse()
 
    And the method to call for the second way of defining an area is:
 
        extract_image_region_interactively_through_mouse_clicks()
 
    Starting with Version 2.0.2, you can now also apply a color filter to
    an input image before it is segmented.  A color is specified by the
    constructor option "color_filter" and the methods to call in your
    script for applying the filter are:
    
        apply_color_filter_hsv()
 
        apply_color_filter_rgb()
 
    The following script in the Examples directory illustrates segmenting
    just a portion on an image and using a color filter:
 
        extract_image_portion_and_apply_color_filter_demo.py        
 
 
THE ImageScanner CLASS:
 
    Starting with Version 2.2.0, you can use all of the functionality of
    the Watershed class through the new ImageScanner class if you want to
    use a scanning approach to solving a computer vision problem.  You will
    find the ImageScanner class particularly useful if you are dealing with
    large images and you want to segment out small objects.  Consider the
    fact that you can now purchase a fairly inexpensive digital camera
    capable of recording 5472x3648 images.  Let us say you use such a
    camera to record images of large scenes and that you want to
    automatically analyze those images for the presence of very small
    objects.  Developing and testing algorithms for such problems is best
    carried out by first scanning the large image using a scanning window
    whose size is commensurate with the size of the objects you are
    interested in.  That is, you first divide the large image into
    overlapping or non-overlapping subimages and you then focus on reliable
    object detection in the subimages.  This is where you'll find the
    ImageScanner class useful.  This class has been provided with methods
    that allow you to see the results of the operations carried out on the
    subimages both in the subimages and the original image simultaneously.
    
    The ImageScanner class is programmed as a subclass of the Watershed
    class.  Therefore, it inherits all of the functionality of the parent
    Watershed class.
    
    Here is now you'd call the constructor of the ImageScanner class for
    creating a dump of the subimages collected from all the positions of a
    scanning window as it moves across the image left to right and top to
    bottom:
    
        import ImageScanner
        image_file = "DSC00780.JPG"
        scanner = ImageScanner.ImageScanner(
                       data_image = image_file,
                       binary_or_gray_or_color = "color",
                       scanner_dump_directory = "scanner_dump",
                       scanning_window_width = 800,
                       scanning_window_height = 800
                  )
        
    After you have created a dump directory of the subimages in this manner,
    you'll call the ImageScanner constructor in the following fashion for
    object detection:
    
        import ImageScanner
        image_file = "DSC00780.JPG"
        scanner = ImageScanner.ImageScanner(
                       data_image = image_file,
                       binary_or_gray_or_color = "color",
                       scanner_dump_directory = "scanner_dump",
                       scanning_window_width = 800,
                       scanning_window_height = 800,
                       color_filter = [(0,30),(0,255),(0,255)],
                       min_brightness_level = 100,
                       min_area_threshold = 800,               
                       max_area_threshold = 8000,
                       max_number_of_blobs = 20,
                       object_shape  = "circular",
                 )
        positions_of_detected_objects = scanner.analyze_scanner_dump()
        
    As to what shapes would be detected by this kind of a constructor call
    depends on the code you place in the method analyze_scanner_dump() of
    the ImageScanner class.  The code shown for that method in the module
    is just a place holder for the kind of logic you would need to place in
    that method.
    
 
CONSTRUCTOR PARAMETERS OF THE ImageScanner CLASS:
 
    Since the ImageScanner class is a subclass of the Watershed class, you
    can supply its constructor with any parameter that is recognized by the
    parent Watershed class.  Shown below is a list of parameters that are
    specific to the ImageScanner class:
 
    max_number_of_blobs: The maximum number of blobs you would like the
                    module to examine in each subimage for detecting the
                    object of interest.
 
    max_area_threshold: If you have a rough sense of the number of pixels
                    an instance of the object you are trying to detect will
                    occupy, you can use this threshold to discard blobs
                    that are significantly larger.
 
    min_area_threshold: As with the previous parameter, if you have a rough
                    sense of the number of pixels an instance of the object
                    you are trying to detect will occupy, you can use this
                    parameter to discard blobs that are significantly
                    smaller.
 
    min_brightness_level: This is a useful parameter in order to experiment
                    with the suppression of noise induced false-alarm
                    detections.
 
    object_shape: This characterizes the shape of the objects you want to
                    detect.  The code shown in the module is for the
                    "circular" shape.  This is just to serve as an example
                    of how to use the module.  
 
    scanner_dump_directory: When scanning a large image with a scanning
                    window, the subimages produced at all positions of the
                    scanning window are dumped in the directory named by
                    this parameter.
 
    scanning_window_height: As should be obvious by its name, this
                    parameter specifies the height of the scanning window.
 
    scanning_window_width: As with the previous entry, this parameter
                    specifies the width of the scanning window.
 
 
PUBLIC METHODS OF THE ImageScanner CLASS:
 
    (1) analyze_scanner_dump():
 
        The purpose of this method is to analyze ALL images in the scanner
        dump directory, generate the object detection results from each
        subimage in the directory, and, finally, display the locations of
        the detections collected from all the subimages in the original
        image.
 
 
    (2) analyze_scanner_dump_and_show_intermediate_results():
 
        Like the previous method, the purpose of this method is to analyze
        ALL images in the scanner dump directory, generate the object
        detection results from each subimage in the directory, and,
        finally, display the locations of the detections collected from all
        the subimages in the original image.  This method will also display
        the intermediate results obtained as the subimages are processed
        for object detections.  The intermediate results correspond to
        color filtering, binarizing, component labeling, etc.
 
 
    (3) analyze_single_subimage_from_image_scan():
 
        The algorithm development for detecting small objects in a large
        image is best carried out by working with just the subimages in
        which it is easier to see the objects.  This method does exactly
        that.  As you would expect, when invoking this method, your
        constructor call for the ImageScanner class must name the subimage
        you are interested in.
 
 
    (4) analyze_single_subimage_from_image_scan_and_show_intermediate_results():
 
        This method does the same thing as the previous method, except that
        it also shows you the intermediate results obtained from the named
        subimage.  The intermediate results correspond to color filtering,
        binarizing, component labeling, etc.
 
 
    (5) scan_image_for_detections_interactive():
 
        Before you can use the functionality of the ImageScanner class for
        anything useful, you must execute this method (or the method named
        next) in order to create a dump of the subimages extracted for the
        large input image.  This method is interactive in the sense that it
        shows you each subimage in its own popup window before depositing
        it in the dump directory.  Since it is much easier to see small
        objects in the subimages displayed in this manner, this mode of
        scanning a large image help you better understand the object
        detection problem you are dealing with.
 
 
    (6) scan_image_for_detections_noninteractive():
 
        This method does exactly the same thing as the previous method ---
        it creates a directory dump of the subimages extracted from the
        large input image.  It is faster because it is not interactive.
        That is, you are NOT shown the subimages before they are deposited
        in the dump directory.
       
 
THE Examples DIRECTORY:
 
    See the Examples directory in the distribution for the different ways
    in which you can use this module.  If you just want to play with
    dilate-erode methods of the module, execute the script
 
        dilate_erode.py
 
    This script assumes a disk structuring element whose radius and shape
    are supplied as the arguments to the methods `dilate(radius,shape)' and
    `erode(radius,shape)'.  To demonstrate the usefulness of these
    operations for "repairing" breaks in edges, execute the script
 
        edge_repair.py
 
    If you want to play with the distance mapping code in the module,
    execute the script:
    
        distance_mapping.py
 
    This script will ask you to place a mark with a mouse click in one of
    the blobs in your binary image.  Subsequently, it will display a
    distance map of the blob with respect to that mark.  For a
    demonstration that involves more complex blobs --- these being blobs
    with holes in them --- execute the script
 
        distance_mapping2.py
 
    For a demonstration of the calculation of the influence zones (IZ) in a
    binary blob, execute the script
 
        influence_zones.py
 
    For a visually interesting demonstration of IZ calculation, you must
    place at least two marks inside a blob.  Each mark is dilated into its
    IZ and the boundaries between the IZs constitute the geodesic skeleton
    of the binary blob with respect to the marks placed in the blob.
 
    All of the scripts mentioned above run on binary image files.  As a
    first demonstration involving grayscale or color images, execute the
    script
 
        LoG.py
 
    that calculates the Laplacian-of-Gaussian of an image.  The LoG is
    calculated by taking a difference of two Gaussian-smoothed images with
    two different values of sigma.  The first Gaussian smoothed image is
    calculated with the sigma as set in the constructor and the second with
    a sigma that is 20% larger.
 
    To see an automatic watershed segmentation that does NOT involve any
    user interaction, execute the script:
 
        segment_automatic_and_show_watershed.py
 
    As you will notice, when there is no help from the user, the watershed
    algorithm over-segments the image.  For an example of the segmentation
    produced by this script, for the following image
 
        orchid0001.jpg
 
    of an orchid, the script produced the segmentation shown in
 
        automatic_output_segmentation_for_orchid.jpg
 
    To see the individual blobs extracted from your input image through the
    watershed contours, execute the following script:
 
        segment_automatic_and_use_contours_to_extract_blobs.py
 
    This script finds the blobs by using the logic that a pixel belongs to
    the region bounded by a contour if a line through the pixel intersects
    the contour an even number of times. For a totally different approach
    to blob extraction, you may wish to try the script:
 
        segment_automatic_and_use_region_growing_to_extract_blobs.py
 
    This script uses region-growing logic to pull out the individual blobs.
 
    That brings us to marker based watershed segmentation in which a user
    is asked to place marker points in an image in order to manually modify
    the gradient map. To see how this works, execute the script:
 
        segment_with_markers_and_show_watershed.py
 
    In order to interact with the module for this segmentation exercise,
    pay careful attention to the titles of the image frames that are
    displayed.  When it asks you to click on the "save" and then "exit"
    buttons that appear at the bottom of the image window, you must do
    exactly do that (as opposed to just closing the window).  To see
    all the markers I placed in the image in one of my own attempts at
    segmenting the orchid image, view the image file:
 
       composite_image_with_all_marks_orchid.jpg
 
    The watersheds produced by this marker-assisted segmentation can be
    seen in the output image:
 
       marker_assisted_segmentation_for_orchid.jpg
 
    To see the individual blobs extracted for the case of marker-based
    watershed segmentation, execute the following script:
 
        segment_with_markers_and_use_contours_to_extract_blobs.py
 
    As for the "automatic" version of this script, this script finds the
    blobs by using the logic that a pixel belongs to the region bounded by
    a contour if a line through the pixel intersects the contour an even
    number of times. For a totally different approach to blob extraction,
    you may wish to try the script:
 
        segment_with_markers_and_use_region_growing_to_extract_blobs.py
 
    This script uses region-growing logic to pull out the individual blobs.
 
    The script in the Examples directory:
 
        extract_image_portion_and_apply_color_filter_demo.py
 
    illustrates the methods you can invoke to specify just a portion of the
    image that should be subject to watershed segmentation.  There are two
    different methods that you can call on in order to specify the image
    portion that you want segmented: (1) You can click at a point and then
    drag the mouse to define a rectangular portion of the image; (2) You
    can specify any polygonal shaped area by clicking the mouse at the
    vertices of the polygon you have in mind.
 
    The script named above also illustrates how to apply a color filter to
    an image before it is subject to watershed segmentation. The module
    gives you two different methods for applying a color filter: You can
    apply a filter to the HSV representation of the color, or its RGB
    representation.  
 
    Two additional scripts in the Examples directory that you may find
    useful are:
 
        hist_equalize_demo.py
 
        otsu_threshold_hue_demo.py
 
    The first of these illustrates histogram equalization applied to the
    'v' component to the HSV representation of a color image.  If the image
    to which you want to apply watershed segmentation is generally too dark
    or generally too bright, you might consider first histogram equalizing
    it before subjecting it to segmentation.  The second script,
    'otsu_threshold_hue_demo.py' illustrates the power of hue filtering.
    There exist problem domains where a simple hue filter can give you
    useful foreground/background separation.
 
    Finally, if you want to create your own binary images for some of the
    scripts mentioned above, execute the script
 
        data_gen.py
 
    Do not forget to execute the script
 
        cleanup.py
 
    in the Examples directory after running the scripts mentioned above to
    cleanup the intermediate images created by the scripts.  Ordinarily,
    the destructor of the class would take care of such cleanup.  But
    depending on how you exit the module, that may not always happen.
 
 
THE ExamplesImageScanner DIRECTORY:
 
    This directory contains the following scripts:
 
 
    1)  RunImageScanner.py
 
       Given a new camera image in which you want to detect small objects,
       the first thing you must do is to run this script (or the next
       script) to divide the large image into smaller subimages.
 
       This scripts deposits all the subimages in a directory whose
       pathname you specify in the call to the ImageScanner constructor.
 
       This script is non-interactive --- in the sense that it does not let
       you visually examine each subimage before it is written out to the
       dump directory.
 
 
    2)  RunImageScannerInteractive.py
 
       This is the interactive version of the previous script.  Before a
       subimage is written out to the dump directory, you can examine it
       visually for as long as you want in order to "understand" your data
       for the presence or the absence of small objects.
 
       Note that we are talking about objects that may be so small that
       they may not be clearly visible in the original image if shown in
       the small screen of a typical laptop.
 
       It is easier to such objects in the subimages since they are shown
       at a scale that is much larger than how they appear in the overall
       original image.
 
 
    3)  AnalyzeSingleSubimage.py
 
       After you have created a scanner dump directory of subimages using
       either of the previous two scripts, you can call this script to
       debug your object detection logic on a single subimage from that
       directory.
 
       The object detection results produced from the specified subimage
       are shown in the original image to give you a good sense of where in
       the original image you are looking and what the object detections
       look like there.
 
 
    4)  AnalyzeSingleSubimageShowIntermediateResults.py 
 
       This script does the same thing as the previous script, except that
       it also shows the intermediate results in the processing of the
       designated subimages.  These intermediate results include the
       outputs produced by color filtering, by binarizing the data, by the
       connected components algorithm, etc.
           
 
    5)  AnalyzeScannerDump.py
 
       Assuming you have perfected your object detection logic, after you
       have created a subimage dump directory with either of the first two
       scripts mentioned above, you can call this script to process ALL of
       the subimages in the dump directory for object detection.
 
 
    6)  AnalyzeScannerDumpShowIntermediateResults.py
          
       This does the same thing as the previous script, except that it also
       shows you the intermediate results as each subimage is processed.
       These intermediate results include the outputs produced by color
       filtering, by binarizing the data, by the connected components
       algorithm, etc.
           
 
CAVEATS:
 
    As mentioned earlier, this module is NOT meant for production work ---
    meaning situations where the primary goal is just to get good-quality
    segmentations quickly based solely on user-supplied seeds.  For that
    type of work, you should use the OpenCV implementation.  Being pure
    Python, this module is slow compared to the OpenCV implementation.
    However, in my opinion, this module makes it easier to experiment with
    different approaches for implementing the various steps that go into
    the BLM algorithm for a watershed based segmentation of images.
 
 
BUGS:
 
    Please notify the author if you encounter any bugs.  When sending
    email, please place the string 'Watershed' in the subject line to get
    past the author's spam filter.
 
 
ACKNOWLEDGMENTS:
 
    I wrote the code for the ImageScanner class (added to the Watershed
    module in Version 2.2.0) as a part of my collaboration with Dr. Peter
    Hirst, Professor of Horticulture, Purdue University.  We are interested
    in predicting the yield of apple trees by estimating the number of
    fruitlets on the trees in early spring.  Peter is also the source of
    the image DSC00780.JPG that you will find in the ExamplesImageScanner
    directory.
 
 
ABOUT THE AUTHOR:
 
    The author, Avinash Kak, recently finished a 17-year long "Objects
    Trilogy Project" with the publication of the book "Designing with
    Objects" by John-Wiley. If interested, visit his web page at Purdue to
    find out what this project was all about. You might like "Designing
    with Objects" especially if you enjoyed reading Harry Potter as a kid
    (or, as an adult, for that matter).
 
    For all issues related to this module, contact the author at
    kak@purdue.edu
 
    If you send email, please place the string "Watershed" in your
    subject line to get past the author's spam filter.
 
COPYRIGHT:
 
    Python Software Foundation License
 
    Copyright 2018 Avinash Kak
 
@endofdocs

 
Imported Modules
       
PIL.Image
PIL.ImageDraw
PIL.ImageTk
tkinter
copy
functools
glob
math
numpy
os
random
re
signal
sys

 
Classes
       
builtins.object
Watershed

 
class Watershed(builtins.object)
     Methods defined here:
__del__(self)
# The destructor:
__init__(self, *args, **kwargs)
Initialize self.  See help(type(self)) for accurate signature.
apply_color_filter_hsv(self)
apply_color_filter_hsv_to_named_image_file(self, filename, show_intermediate_results=True)
apply_color_filter_rgb(self)
apply_otsu_threshold_to_hue(self, polarity=1, area_threshold=0, min_brightness_level=100)
Applies the optimum Otsu threshold to hue in order to separate the image into foreground and 
background.  Regarding the argument 'polarity', if it is '-1', all pixels whose hue is below the 
optimum Otsu threshold are accepted.  On the other hand, if the polarity is '1', all pixels whole 
hue is higher than the calculated threshold are accepted.  The 'area_threshold' parameter is
passed to the method "connected_components_for_binary_array()" for rejecting blobs that are
smaller than the value supplied through 'area_threshold'.
compute_LoG_image(self)
This method computes the Laplacian-of-Gaussian (LoG) of an image. The LoG is 
calculated as the difference of two Gaussian-smoothed versions of the input 
image at two slightly difference scales.  The LoG itself is NOT used in 
watershed calculations.
compute_Z_level_sets_for_gradient_image(self)
For any value of n between 0 and 255, both ends inclusive, a pixel is in the
set Z_n if the gradient value at that pixel is less than or equal to n.  Note
that the gradient values are normalized to be between 0 and 255.
compute_gradient_image(self)
The Watershed algorithm is applied to the gradient of the input image.  This
method computes the gradient image.  The gradient calculation is carried out
after the image is smoothed with a Gaussian kernel whose sigma is set in the
constructor.
compute_influence_zones_for_marks(self)
Calculates the influence zones in a binary blob with respect to the marks
placed inside the blob.  The method also identifies the pixels at the
geodesic skeleton formed by the influence zones.
connected_components(self, data_or_marks)
This method is the basic connected components algorithm in the Watershed
module.  Just for programming convenience related to the I/O from this
method, I have made a distinction between carrying out a connected-components
labeling of a binary image and doing the same for a binary pattern that
contains all of the marks made by the user.
connected_components_of_filtered_output(self, argimage, min_brightness_level, min_area_threshold, max_area_threshold, expected_number_of_blobs, blob_shape, show_intermediate_results=True)
dilate(self, structuring_element_rad, structuring_ele_shape)
This is to just demonstrate the basic idea of dilation of a binary pattern by
a disk structuring element whose radius is supplied as the first
argument. The precise shape of the structuring element, which can be either
"square" or "circular", is supplied through the second argument. This method
itself is NOT used in the watershed calculations.  For large binary patterns,
it would be more efficient to carry out the dilations only at the border
pixels.
dilate_mark_in_its_blob(self, mark_index)
This method illustrates distance mapping of a blob in a binary image with
respect to a mark created by clicking at a point within the blob.
displayImage(self, argimage, title='')
Displays the argument image.  The display stays on for the number of seconds
that is the first argument in the call to tk.after() divided by 1000.
displayImage2(self, argimage, title='')
Displays the argument image.  The display stays on until the user closes the
window.  If you want a display that automatically shuts off after a certain
number of seconds, use the previous method displayImage().
displayImage3(self, argimage, title='')
Displays the argument image (which must be of type Image) in its actual size.  The 
display stays on until the user closes the window.  If you want a display that 
automatically shuts off after a certain number of seconds, use the method 
displayImage().
displayImage4(self, argimage, title='')
Displays the argument image (which must be of type Image) in its actual size without 
imposing the constraint that the larger dimension of the image be at most half the 
corresponding screen dimension.
displayImage5(self, argimage, title='')
This does the same thing as displayImage4() except that it also provides for
"save" and "exit" buttons.  This method displays the argument image with more 
liberal sizing constraints than the previous methods.  This method is 
recommended for showing a composite of all the segmented objects, with each
object displayed separately.  Note that 'argimage' must be of type Image.
displayImage6(self, argimage, title='')
This does the same thing as displayImage3() except that it also provides for
"save" and "exit" buttons. Note that 'argimage' must be of type Image.
display_all_segmented_blobs(self)
display_data_image(self)
This is just a convenience method for displaying the image that you want to
subject to watershed segmentation.
display_watershed(self)
Displays the watershed segmentation of the image in the grayscale mode.  That
is, the image shown is what the computations are carried out on --- a
grayscale version of the input image (assuming it was a color image).
display_watershed_contours_in_color(self)
Shows the watershed contours as extracted by the extract_watershed_contours()
method.
 
To write the result image as a jpeg file to the disk, click on the "save"
and "exit" buttons at the bottom of the window --- as opposed to just closing
the window.  You can subsequently display the image from the disk file by
using, say, the 'display' function from the ImageMagick library.  You can
create a hardcopy version of the result by sending the jpeg image to a
printer.
display_watershed_in_color(self)
Displays the watershed segmentation on top of the original color image
(assuming that the input image was a color image to begin with.)
 
To write this result image as a jpeg file to the disk, click on the "save"
and "exit" buttons at the bottom of the window --- as opposed to just closing
the window.  You can subsequently display the image from the disk file by
using, say, the 'display' function from the ImageMagick library.  You can
create a hardcopy version of the result by sending the jpeg image to a
printer.
erode(self, argimage, structuring_element_rad, structuring_ele_shape)
This is to just demonstrate the basic idea of erosion of a binary pattern by
a disk structuring element whose radius is supplied as the first argument.
The precise shape of the structuring element, which can be either "square" or
"circular", is supplied through the second argument. This method itself is
NOT used in the watershed calculations.
extract_data_pixels(self)
Gets the binary, grayscale, and color images ready for watershed processing.
If the images are too large, they are reduced to the size set by the
constructor.  Color images are converted into grayscale images.
extract_image_region_interactively_by_dragging_mouse(self)
This is one method you can use to apply watershed segmentation to just a portion of your image.
This method extract the portion you want.  You click at the upper left corner of the rectangular
portion of the image you are interested in and you then drag the mouse pointer to the lower 
right corner.  Make sure that you click on "save" and "exit" after you have delineated the area.
extract_image_region_interactively_through_mouse_clicks(self)
This method allows a user to use a sequence of mouse clicks in order to specify a region of the 
input image that should be subject to watershed segmentation.  The mouse clicks taken together 
define a polygon. The method encloses the polygonal region by a minimum bounding rectangle,
which then becomes the new input image for the rest of processing.
extract_rectangular_masked_segment_of_image(self, horiz_start, horiz_end, vert_start, vert_end)
Keep in mind the following convention used in the PIL's Image class: the first
coordinate in the args supplied to the getpixel() and putpixel() methods is for
the horizontal axis (the x-axis, if you will) and the second coordinate for the
vertical axis (the y-axis).  On the other hand, in the args supplied to the
array and matrix processing functions, the first coordinate is for the row
index (meaning the vertical) and the second coordinate for the column index
(meaning the horizontal).  In what follows, I use the index 'i' with its
positive direction going down for the vertical coordinate and the index 'j'
with its positive direction going to the right as the horizontal coordinate. 
The origin is at the upper left corner of the image.
extract_segmented_blobs_using_contours(self, min_contour_length=50)
This method uses the inside/outside logic at a point (based on the number of times a
line through the point intersects the contour) in order to decide whether the point is
inside a closed contour or outside it.
extract_segmented_blobs_using_region_growing(self, num_of_segments, min_area=50)
This method uses region growing to extract segmented blobs.  We start at a pixel
and grow from there by incorporating other neighboring pixels recursive until we
hit a watershed ridge.
extract_watershed_contours_separated(self)
This method uses the border following algorithm to extract the watershed
contours from the final propagation of influences by the propagate_influences
method.
extract_watershed_contours_with_random_sampling(self, num_of_contours, min_length=20)
This method uses the border following algorithm to extract the watershed
contours from the final propagation of influences by the propagate_influences
method.
histogram_equalize(self)
Carries out contrast enhancement in an image.  If an image is generally too dark or generally too
bright, you may wish to histogram equalize it before subjecting it to watershed segmentation.
is_blob_circular(self, list_of_pixel_coords)
Tests the circularity predicate by taking the ratio of the area of the blob, as measured by
the number of pixels in the blob, to \pi*r^2 where r is its average radius.  This is
obviously much too simple a predicate at this time.  However, you can add to its power
by incorporating additional logic here.
mark_blobs(self, purpose)
For demonstrations of distance mapping of a binary blob with respect to a
marker blob, this method allows a user to both select one or more blobs in a
binary image for the purpose of distance mapping and to also place marks on
the blobs.
mark_blobs_no_image_scale_change(self)
For demonstrations of distance mapping of a binary blob with respect to a
marker blob, this method allows a user to both select one or more blobs in a
binary image for the purpose of distance mapping and to also place marks on
the blobs.
mark_image_regions_for_gradient_mods(self)
For watershed segmentation that incorporates user-supplied modifications to
the image gradients, this method allows a user to demarcate through mouse
clicks polygonal regions where the gradient can be explicitly set to 0.  For
each region thus demarcated, the mouse clicks must be supplied in a clockwise
fashion.
modify_gradients_with_marker_minima(self)
After a user has demarcated the regions in which the image gradients can be
modified, this method carries out the gradient modifications.
propagate_influence_zones_from_bottom_to_top_of_Z_levels(self)
Basic to watershed computation is the calculation of influence zones of the
connected components for one Z level in the connected components in the next
Z level.  Note that we stop at one level below the max level at which Z sets
are known.  That is because the last IZ calculation consists of finding the
influence zones of the Z sets at the 'self.max_grad_level-1' level in the Z
sets at the 'self.max_grad_level' level.
Static methods defined here:
gendata(feature, imagesize, position, orientation_or_radius, output_image_name)
This method is useful for generating simple binary patterns for checking the
validity of the logic used for dilation, erosion, IZ calculation, geodesic
skeleton calculation, etc.  Note that the permissible values for the
'feature' parameter are: 'line', 'triangle', 'rectangle', 'broken_rectangle',
'disk' and 'twin_rect'.  The parameter 'imagesize' is supposed to be a tuple
(m,n) for the size of the output image desired.  The parameter 'position' is
supposed to be a tuple (x,y) of pixel coordinates for specifying the position
of the binary pattern in your image.  The parameter 'orientation_or_radius' is an
integer value specifying the number of degrees of rotation that should be
applied to the pattern for a given 'feature' for the case of 'line', 'triangle',
'rectangle' and 'broken_rectangle'.  The same parameter for the case of 'circle'
means the radius of the disk.  Note that the x coordinate is along the horizontal 
direction pointing  to the right and the y coordinate is along vertical direction 
pointing downwards.  The origin is at the upper left corner.
make_binary_pic_art_nouveau(under_what_name)
Can be used to make "fun" binary images for demonstrating distance mapping of
binary blobs, calculation of influence zones, etc.  This method is taken from
Chapter 13 of my book "Scripting with Objects".
Data descriptors defined here:
__dict__
dictionary for instance variables (if defined)
__weakref__
list of weak references to the object (if defined)
Data and other attributes defined here:
canvas = None
drawEnable = 0
region_mark_coords = {}
startX = 0
startY = 0

 
Functions
       
ctrl_c_handler(signum, frame)
file_index(file_name)
This function is needed if you are analyzing an image in the scanning mode.
In this mode, the module runs a non-overlapping scanning window over the image,
dumps the image data inside the scanning window at each position of the window
in a directory.  Each file is given a name that includes a numerical index
indicating its position in the original images.  This function returns the
numerical index in the name of a file.

 
p
        __author__ = 'Avinash Kak (kak@purdue.edu)'
__copyright__ = '(C) 2020 Avinash Kak. Python Software Foundation.'
__date__ = '2020-October-5'
__url__ = 'https://engineering.purdue.edu/kak/distWatershed/Watershed-2.2.2.html'
__version__ = '2.2.2'
 
Author
        Avinash Kak (kak@purdue.edu)