コード例 #1
0
ファイル: goban.py プロジェクト: Tartarichi/Rocamgo-ng
    def search_stones_mask(self, image, threshold):
        stones = []
        smooth = cv.CloneImage(image)
        cv.Smooth(image, smooth, cv.CV_GAUSSIAN, 5, 5)
        hsv_img = cv.CreateImage(cv.GetSize(image), 8, 3)
        cv.CvtColor(smooth, hsv_img, cv.CV_RGB2HSV)

        masked_img = cv.CreateImage(cv.GetSize(hsv_img), 8, 1)
        #FIXME: Illumination has a strong effect in white detection, try to fix
        # it by tweaking the threshold or reduce contrast in the image
        trickeryfu = {WHITE: [(0, 0, 191), (180, 255, 255)],
            BLACK: [(0, 0, 0), (180, 255, 64)]}
        for k in trickeryfu.keys():
            color_range = trickeryfu.get(k)
            cv.InRangeS(hsv_img, color_range[0], color_range[1], masked_img)
            circles = self._get_circles(masked_img)
            for n in range(circles.cols):
                # TODO: Try to make this less ugly
                pixel = cv.Get1D(circles, n)
                pt = (cv.Round(pixel[0]), cv.Round(pixel[1]))
                radius = cv.Round(pixel[2])
                position = Move.pixel_to_position(image.width, pt)
                stones.append(Move(k, position))
                cv.Circle(image, pt, radius, cv.CV_RGB(255, 255, 255)
                    if k == BLACK else cv.CV_RGB(0, 0, 0), 2)
        return image, stones
コード例 #2
0
ファイル: goban.py プロジェクト: Tartarichi/Rocamgo-ng
    def search_stones_simple(self, image, threshold):
        stones = []
        smooth = cv.CloneImage(image)
        cv.Smooth(image, smooth, cv.CV_GAUSSIAN, 5, 5)
        hsv_img = cv.CreateImage(cv.GetSize(image), 8, 3)
        cv.CvtColor(smooth, hsv_img, cv.CV_RGB2HSV)

        masked_img = cv.CreateImage(cv.GetSize(hsv_img), 8, 1)
        trickeryfu = {WHITE: [(0, 0, 191), (180, 255, 255)],
            BLACK: [(0, 0, 0), (180, 255, 64)]}
        for k in trickeryfu.keys():
            color_range = trickeryfu.get(k)
            cv.InRangeS(hsv_img, color_range[0], color_range[1], masked_img)
            storage = cv.CreateMemStorage()
            contours = cv.FindContours(masked_img, storage,
                cv.CV_RETR_EXTERNAL, cv.CV_CHAIN_APPROX_SIMPLE,
                offset=(0, 0))
            while contours:
                # The original idea was to find the centroids.
                #   This seems simpler
                perimeter = cv.ArcLength(contours)
                expected_perim = math.pi * image.width / GOBAN_SIZE
                if (perimeter < 1.1 * expected_perim and
                    perimeter > 0.9 * expected_perim):
                    __, center, radius = cv.MinEnclosingCircle(contours)
                    position = Move.pixel_to_position(image.width, center)
                    stones.append(Move(k, position))
                    center = tuple(cv.Round(v) for v in center)
                    radius = cv.Round(radius)
                    cv.Circle(image, center, radius, cv.CV_RGB(255, 255, 255)
                        if k == BLACK else cv.CV_RGB(0, 0, 0), 2)
                contours = contours.h_next()
        return image, stones
コード例 #3
0
ファイル: goban.py プロジェクト: Tartarichi/Rocamgo-ng
 def search_stones_old(self, image, threshold):
     circles = search_stones(image, None)
     false_stones = 0
     stones = []
     for n in range(circles.cols):
         pixel = cv.Get1D(circles, n)
         pt = (cv.Round(pixel[0]), cv.Round(pixel[1]))
         radious = cv.Round(pixel[2])
         # Comprobar el color en la imagen
         color = check_color_stone(pt, radious, image, threshold)
         position = Move.pixel_to_position(image.width, pixel)
         if color == BLACK:
             cv.Circle(image, pt, radious, cv.CV_RGB(255, 0, 0), 2)
             stones.append(Move(color, position))
         elif color == WHITE:
             cv.Circle(image, pt, radious, cv.CV_RGB(0, 255, 0), 2)
             stones.append(Move(color, position))
         else:
             #cv.Circle(image, pt, radious, cv.CV_RGB(255,255,0), 2)
             false_stones += 1
     return image, stones
コード例 #4
0
ファイル: goban.py プロジェクト: Tartarichi/Rocamgo-ng
 def search_stones_LaB(self,image,th):
     stones = []
     false_stones = 0
     circles = search_stones(image, None)
     lab_img = cv.CreateImage(cv.GetSize(image), 8, 3)
     cv.CvtColor(image, lab_img, cv.CV_BGR2Lab)
     for n in range(circles.cols):
         pixel = cv.Get1D(circles, n)
         pt = (cv.Round(pixel[0]), cv.Round(pixel[1]))
         radious = cv.Round(pixel[2])
         color = check_color_stone_LaB(pt, radious, lab_img)
         position = Move.pixel_to_position(image.width, pixel)
         if color == BLACK:
             # print "BLACK"
             cv.Circle(image, pt, radious, cv.CV_RGB(255, 0, 0), 2)
             stones.append(Move(color, position))
         elif color == WHITE:
             # print "WHITE"
             cv.Circle(image, pt, radious, cv.CV_RGB(0, 255, 0), 2)
             stones.append(Move(color, position))
         else:
             false_stones += 1
     return image, stones
コード例 #5
0
ファイル: goban.py プロジェクト: Virako/Rocamgo-ng
 def search_stones_old(self, image, threshold):
     circles = search_stones(image)
     false_stones = 0
     stones = []
     if circles == None:
         return image, stones
     for ci in circles[0,:]:
         ci = np.rint(ci).astype(np.int) # apply round and int
         pt = (ci[0], ci[1])
         radious = ci[2]
         # Comprobar el color en la imagen
         color = check_color_stone(pt, radious, image, threshold)
         position = Move.pixel_to_position(image.shape[0], pt)
         if color == BLACK:
             circle(image, pt, radious, (255, 255, 255), 2)
             stones.append(Move(color, position))
         elif color == WHITE:
             circle(image, pt, radious, (0, 0, 0), 2)
             stones.append(Move(color, position))
         else:
             circle(image, pt, radious, (255,0,0), 2)
             false_stones += 1
     return image, stones
コード例 #6
0
ファイル: goban.py プロジェクト: Virako/Rocamgo-ng
 def search_stones_LaB(self,image,th):
     stones = []
     false_stones = 0
     circles = search_stones(image)
     lab_img = np.zeros(image.shape[:2], np.uint8)
     lab_img = cvtColor(image, COLOR_BGR2Lab)
     for n in range(circles.cols):
         pixel = Get1D(circles, n)
         pt = (Round(pixel[0]), Round(pixel[1]))
         radious = Round(pixel[2])
         color = check_color_stone_LaB(pt, radious, lab_img)
         position = Move.pixel_to_position(image.width, pixel)
         if color == BLACK:
             # print("BLACK")
             circle(image, pt, radious, (255, 0, 0), 2)
             stones.append(Move(color, position))
         elif color == WHITE:
             # print("WHITE")
             circle(image, pt, radious, (0, 255, 0), 2)
             stones.append(Move(color, position))
         else:
             false_stones += 1
     return image, stones