example3_2_process_training_set.py

import numpy as np
import cv2
from matplotlib import pyplot as plt

for im_id in range(1,4):
    print(im_id)
    im = cv2.imread("SkinTrain"+str(im_id)+".jpg")
    mask = cv2.imread("SkinTrain"+str(im_id)+"_mask.jpg",0)

    im_hsv = cv2.cvtColor(im,cv2.COLOR_BGR2HSV)
    h = im_hsv[:,:,0]
    s = im_hsv[:,:,1]

    h_skin = [val for (i, val) in enumerate(h.reshape(1, -1)[0]) if mask.reshape(1, -1)[0][i] >= 128]
    s_skin = [val for (i, val) in enumerate(s.reshape(1, -1)[0]) if mask.reshape(1, -1)[0][i] >= 128]
    h_nonskin = [val for (i, val) in enumerate(h.reshape(1, -1)[0]) if mask.reshape(1, -1)[0][i] < 128]
    s_nonskin = [val for (i, val) in enumerate(s.reshape(1, -1)[0]) if mask.reshape(1, -1)[0][i] < 128]

    if im_id == 1:
        h_skin_all = h_skin
        s_skin_all = s_skin
        h_nonskin_all = h_nonskin
        s_nonskin_all = s_nonskin
    else:
        h_skin_all = np.concatenate((h_skin_all,h_skin))
        s_skin_all = np.concatenate((s_skin_all,s_skin))
        h_nonskin_all = np.concatenate((h_nonskin_all,h_nonskin))
        s_nonskin_all = np.concatenate((s_nonskin_all,s_nonskin))

plt.plot(h_nonskin_all,s_nonskin_all,'b.')
plt.plot(h_skin_all,s_skin_all,'r.')
plt.show()

cv2.waitKey(0)
cv2.destroyAllWindows()