fitushar / skin-lesion-segmentation-using-grabcut

Skin lesion segmentation is one of the first steps towards automatic Computer-Aided Diagnosis of skin cancer. Vast variety in the appearance of the skin lesion makes this task very challenging. The contribution of this paper is to apply a power foreground extraction technique called GrabCut for automatic skin lesion segmentation in HSV color space with minimal human interaction. Preprocessing was performed for removing the outer black border. Jaccard Index was measured to evaluate the performance of the segmentation method. On average, 0.71 Jaccard Index was achieved on 1000 images from ISIC challenge 2017 Training Dataset.

Automatic Skin Lesion Segmentation Using GrabCut in HSV Color Space

Skin lesion segmentation is one of the first steps towards automatic Computer-Aided Diagnosis of skin cancer. Vast variety in the appearance of the skin lesion makes this task very challenging. The contribution of this paper is to apply a power foreground extraction technique called GrabCut for automatic skin lesion segmentation in HSV color space with minimal human interaction. Preprocessing was performed for removing the outer black border. Jaccard Index was measured to evaluate the performance of the segmentation method. On average, 0.71 Jaccard Index was achieved on 1000 images from ISIC challenge 2017 Training Dataset. In this work, a framework was proposed for skin lesion segmentation based on automatic GrabCut segmentation. Auto Extracting mask and rectangle initialization strategies was shown for making the segmentation algorithm automatic and generic. The algorithm achieved over 0.71 average Jaccard index for 1000 test images. Future work will be focused on exploring different color channels to improve the performance.

Qualitative and Quantitative results of segmentation

skin-lesion-segmentation-using-grabcut's Issues

Error on line 'cv2.grabCut(img,mask,None,bgdModel,fgdModel,5,cv2.GC_INIT_WITH_MASK)'

Hello,

I am trying to use your code for the skin leison segmentation and I am getting the following error:

error                                     Traceback (most recent call last)
<ipython-input-19-daaa4cb8cf6e> in <module>
    119         mask[newmask == 255] = 1
    120 
--> 121         dim= cv2.grabCut(img,mask,None,bgdModel,fgdModel,5,cv2.GC_INIT_WITH_MASK)
    122         mask2 = np.where((mask==2)|(mask==0),0,1).astype('uint8')
    123         GrabCut_img = img*mask2[:,:,np.newaxis]

error: OpenCV(4.2.0) C:\projects\opencv-python\opencv\modules\imgproc\src\grabcut.cpp:386: error: (-215:Assertion failed) !bgdSamples.empty() && !fgdSamples.empty() in function 'initGMMs'

The image and code are in the right folder. Do you know why am I getting this error? The Green_Mask and the Inverse_Green_mask for the image are empty, and enhanced_bgr_image looks like this:

The original image is: