GreenThumb CV Leaf Area Filter

leaf_area.py

import numpy as np
import imutils
import cv2
import matplotlib.pyplot as plt
%matplotlib inline  

# Defined bounds using https://github.com/jrosebr1/imutils/blob/master/bin/range-detector
leaf_lower = (29, 0, 0)
leaf_upper = (255, 255, 128)

def leaf_mask(bgr_image):
    frame = imutils.resize(bgr_image, width=500)
    blurred = cv2.GaussianBlur(frame, (11, 11), 0)
    hsv = cv2.cvtColor(blurred, cv2.COLOR_BGR2HSV)

    # construct a mask for the color "green", then perform
    # a series of dilations and erosions to remove any small
    # blobs left in the mask
    mask = cv2.inRange(hsv, leaf_lower, leaf_upper)
    mask = cv2.erode(mask, None, iterations=4)
    mask = cv2.dilate(mask, None, iterations=2)
    return mask
  
v2.countNonZero(week_1_mask)
week_1_mask = leaf_mask(week_1)
plt.title('Week 1 Leaf Area')
week_1_image = imutils.resize(week_1, width=500)
week_1_output = cv2.bitwise_and(week_1_image, week_1_image, mask = week_1_mask)