Exemple #1
0
def refineLabels(out, labels, corners, howmany=3):
    # pick up labels on corners
    containcorners = []
    for corner in corners:
        x = int(corner[0][0])
        y = int(corner[0][1])
        hsv = rgb2hsv(out[y, x])
        # accept only if candidate color(yellow or blue)
        if checkBlue(hsv) or checkYellow(hsv):
            containcorners.append(labels[y][x])
    # make table(corners,label)
    match = []
    for label in np.unique(containcorners):
        match.append([(containcorners == label).sum(), label])

    match = sorted(match, reverse=True)
    match = np.array(match)
    # refine howmany
    howmany = howmany if howmany < match.shape[0] else match.shape[0]
    # select labels with enough corners
    pickup = []
    for i in range(howmany):
        pickup.append(match[i][1])
    # mark bg with 0
    mark = 0
    labels_ = np.zeros(labels.shape)
    #
    for label in np.unique(pickup):
        mark += 1
        labels_[labels == label] = mark
    # make saturation

    return labels_
Exemple #2
0
def estimateColr(img,
                 rect,
                 debug=False):
    [x, y, w, h] = rect
    roi = img[y:y+h, x:x+w]
    rgb = cv2.cvtColor(roi,cv2.COLOR_BGR2RGB)
    totalcolr = [0,0,0]
    for j in range(h):
        for i in range(w):
            totalcolr += rgb[j,i]
    pixelcount = w * h
    avergecolr = totalcolr / pixelcount
    if debug:
        print("estimatecolr,avrgcolr:",avergecolr)
    hsv = rgb2hsv(avergecolr)
    return checkBlue(hsv) or checkYellow(hsv)
 def detectcolr(self):
     c_x = self.width / 2
     c_y = self.height / 2
     roi = self.bgr[c_y - 15:c_y + 15, c_x - 30:c_x + 30, :]
     rgb = cv2.cvtColor(roi, cv2.COLOR_BGR2RGB)
     h, w, c = roi.shape
     totalcolr = np.zeros(3)
     for j in range(h):
         for i in range(w):
             totalcolr += rgb[j, i]
     avergecolr = totalcolr / (w * h)
     hsv = rgb2hsv(avergecolr)
     if checkBlue(hsv):
         self.colr = 1
         self.returnImg = 255 - self.laplacian
     elif checkYellow(hsv):
         self.colr = 2
         self.returnImg = self.laplacian.copy()
     else:
         self.colr = 0