feature-detection.py

# import the necessary packages
import cv2
import sys
import os
import numpy as np
import argparse
sys.path.append(os.path.abspath('./modules/'))
import detection
import helpers

ap = argparse.ArgumentParser()
ap.add_argument("-d", "--debug", type=bool, default=False,
    help="experiment with different image filters")
ap.add_argument("-i", "--imagepath", type=str, 
    default = None,
    help="path to test image")
args = ap.parse_args()
if(args.imagepath is None):
    image_path = helpers.get_random_test_image()
else:
    image_path = args.imagepath

image = cv2.imread(image_path,1)
ball = detection.GolfBallDetection(image)
if ball:
    (x,y,w,h) = ball
    x_plus_w = x+w
    y_plus_h = y+h

    
    section = image.copy()[y:y+h, x:x+w]
    
    cv2.imshow("Crop to ball",section)
        
    # generating the kernels
    kernel1 = np.array([[0, -1, -1], # kernel for embossing bottom left side
                        [1, 0, -1],
                        [1, 1, 0]])
                        
    # you can generate kernels for embossing top as well
    gray = cv2.cvtColor(section, cv2.COLOR_BGR2GRAY)
    gray = cv2.GaussianBlur(gray, (7, 7), 0)
    if(args.debug):
        cv2.imshow("Gray",gray)

    
    if(args.debug):
        embossdepth = np.ones((h, w), np.uint8) * 128
        embossed = cv2.add(cv2.filter2D(gray, -1, kernel1),embossdepth) # emboss on bottom left side
        cv2.imshow("Embossed",embossed)

    if(args.debug):
        _, binary = cv2.threshold(cv2.cvtColor(section, cv2.COLOR_BGR2GRAY), 20, 255, cv2.THRESH_BINARY | cv2.THRESH_OTSU)
        binary = 255 - binary
        cv2.imshow("Binary",binary)

    if(args.debug):
        edges = cv2.Canny(binary,100,200)
        cv2.imshow("Edges",edges)

    if(args.debug):
        embossed_edges = cv2.Canny(embossed,100,200)
        cv2.imshow("Embossed Edges",embossed_edges)
        contours,heirarchy = cv2.findContours(embossed_edges, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
        im2 = cv2.cvtColor(section, cv2.COLOR_BGR2GRAY)
        dimples = []
        for contour in contours:
            approx = cv2.approxPolyDP(contour,0.01*cv2.arcLength(contour,True),True)
            area = cv2.contourArea(contour)
            if ((len(approx) > 8) & (area > 30) ):
                dimples.append(contour)

        cv2.drawContours(im2, dimples, -1, (0,255,0), 3)
        cv2.imshow("Contours",im2)

    circle = detection.get_ball_circle(image,x,y,w,h)
    if circle is not None:
        detection.draw_circle_around_ball(image,circle,(x,y))
    else:
        detection.draw_boundaries_and_label(image,(x,y),(w,h),(0,255,0),'Ball')        
else:
    print('[INFO] Failed to find ball')
cv2.imshow("Detecting features",image)

cv2.waitKey(0)
cv2.destroyAllWindows()