Draw a contour around eye-CodePudding

I want to draw the contour around extracted sclera. (Sclera segmentation was done by using this

This is actual output I want to get.

    import numpy as np 
    from cv2 import cv2
    from matplotlib import pyplot as plt
    
    img = cv2.imread('sclera.jpg',1)
    gray = cv2.cvtColor(img,cv2.COLOR_RGB2GRAY)
    cv2.imshow("Adaptive gray Image",gray)
    
    #Contours
    contours, hierarchy = cv2.findContours(gray,cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE) 
    
    img2 = img.copy()
    index = -1 #if negative all contours are drawn, index of contour to be drawn
    thickness = 2
    color = (5,228,72)
    cv2.drawContours(img2,contours,index,color,thickness)
    # cv2.imshow('Contours',img2)
    
    for cnt in contours:
        area = cv2.contourArea(cnt)
        print("Detected Contour with Area: ", area)
    
    cv2.waitKey(0)
    cv2.destroyAllWindows()
    img3 = cv2.cvtColor(img2,cv2.COLOR_RGB2BGR)
    plt.imshow(img3)

Is anybody know why I get the lot of contours in my output? Where could be caused to produce the wrong output?

Any suggestions would appreciate.

CodePudding user response：

Before doing binarization, you need to additionally blur the image to eliminate unnecessary details for which the contour detection algorithm of your choice clings. Use Gaussian Blur to eliminate high-frequency data at the expense of a larger blur radius and amount. Then binarize the image. And only after your image looks like a white spot of the sclera on a black background - only then the contour detection algorithm will give you the correct contour of the sclera.

You can also use the erosion and dilation operations to eliminate many noisy data points and improve contour fit to the shape of the sclera. This should be done after binarization, then you will have a good pre-image for processing by the edge detection algorithm.

In general, you will have the following sequence of actions:

Converting an image to gray
Gaussian blur with a large kernel
Binarization to black and white image
Sequences of erosion-dilation operations
Finding the contour and choosing the largest

CodePudding user response：

As @Egor Zamotaev mentioned in the accepted answer, I have improved the code accordingly to draw the contour, which is given in below.


# Read the sclera segmented image as RGB mode
image = cv2.imread('/content/drive/MyDrive/Research/Testings/sclera.jpg', 1)

# Convert the image to grayscale format
img_gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

# Apply binary thresholding
ret, thresh = cv2.threshold(img_gray, 150, 255, cv2.THRESH_BINARY)

# Visualize the binary image
plt.imshow(cv2.cvtColor(thresh, cv2.COLOR_BGR2RGB))

# Detect the contours on the binary image using cv2.CHAIN_APPROX_NONE
contours, hierarchy = cv2.findContours(image=thresh, mode=cv2.RETR_TREE, method=cv2.CHAIN_APPROX_NONE)
                                    
# Draw contours on the original image
image_copy = image.copy()
cv2.drawContours(image=image_copy, contours=contours, contourIdx=-1, color=(0, 255, 0), thickness=20, lineType=cv2.LINE_AA)
                
# see the results
plt.imshow(cv2.cvtColor(image_copy, cv2.COLOR_BGR2RGB))