Plot Classes

This is a plotting method used to color masks.

plantcv.visualize.colorize_masks(masks, colors)

returns colored_img

  • Parameters:
    • masks - A list of masks to plot
    • colors - A list of colors used to plot the given classes. Users can provide a list of strings from the 36 colors listed below, or list tuples.
  • Context:
    • This function was written to allow users to plot masks returned by naive bayes multiclass but can be used with any mask. The number of masks much match the number of colors.
  • Example use:

Named Colors

Screenshot

Original image

Screenshot


from plantcv import plantcv as pcv

# Run naive bayes multiclass and save a list of masks 
mask = pcv.naive_bayes_classifier(img, pdf_file="machine_learning.txt")

# Plot each class with it's own color 
plotted = pcv.visualize.colorize_masks(masks=[mask['plant'], mask['pustule'], mask['background'], mask['chlorosis']], 
                                       colors=['green', 'red', 'gray', 'gold'])

Masks: (from top to bottom) plant, chlorosis, pustule, background

Screenshot

Screenshot

Screenshot

Screenshot

Plot with Colored Masks

Screenshot