def __getBackGround3(self, threshold, color): """ Method number 4 to detect the background pixels. Detect the background comparing each pixel of local image with the given color using the threshold. (TOO SLOW) @param threshold: Edge to detect the color background @type threshold: int @param color: Color used to detect background @type color: cvScalar @return: As first parameter the original image changing the different pixel to black pixel. As Second parameter a IplImage in Gray Scale where the differences are white pixels. After this can be used as mask. """ def intervale(id1, id2): if math.fabs(id1 - id2) < threshold: return True return False differenceI = cv.CloneMat(self.__image) size = cv.GetSize(self.__image) grayI = cv.CreateImage(cv.GetSize(differenceI), cv.IPL_DEPTH_8U, 1) cv.SetZero(grayI) for i in range(0, size[0]): for j in range(0, size[1]): pixel = cv.Get2D(self.__image, j, i) if intervale(color[0], pixel[0]) and intervale( color[1], pixel[1]) and intervale(color[2], pixel[2]): cv.Set2D(differenceI, j, i, cv.Scalar(0, 0, 0)) cv.Set2D(grayI, j, i, cv.Scalar(255, 255, 255)) cv.Not(grayI, grayI) return differenceI, grayI
def __getBackGround2(self, threshold, preImage): """ Method number 3 to detect the background pixels. The background is detected of the local Image using the given preImage. For this it is calculed the difference between them. @type threshold: int @param preImage: Image used to compare the differences @type preImage: IplImage @return: As first parameter the original image changing the different pixel to black pixel. As Second parameter a IplImage in Gray Scale where the differences are white pixels. After this can be used as mask. """ differenceI = cv.CreateImage(cv.GetSize(self.__image), cv.IPL_DEPTH_8U, 3) grayI = cv.CreateImage(cv.GetSize(differenceI), cv.IPL_DEPTH_8U, 1) cv.AbsDiff(self.__image, preImage, differenceI) cv.CvtColor(differenceI, grayI, cv.CV_BGR2GRAY) cv.Threshold(grayI, grayI, threshold, 255, cv.CV_THRESH_BINARY) cv.Smooth(grayI, grayI, cv.CV_MEDIAN, 5) cv.Smooth(grayI, grayI, cv.CV_GAUSSIAN, 5) cv.SetZero(differenceI) cv.Add(self.__image, differenceI, differenceI, grayI) return differenceI, grayI
def __getBackGround(self, threshold, color): """ Method number 1 to detect the background pixels. The background is detected of the local Image using given color. For this it is calculed the difference between local Image and another image set up the given color only. @param threshold: Edge to detect the color background @type threshold: int @param color: Color used to detect background @type color: cvScalar @return: As first parameter the original image changing the different pixel to black pixel. As Second parameter a IplImage in Gray Scale where the differences are white pixels. After this can be used as mask. """ differenceI = cv.CreateImage(cv.GetSize(self.__image), cv.IPL_DEPTH_8U, 3) grayI = cv.CreateImage(cv.GetSize(self.__image), cv.IPL_DEPTH_8U, 1) backgroundI = cv.CreateImage(cv.GetSize(self.__image), cv.IPL_DEPTH_8U, 3) cv.Set(backgroundI, color) cv.AbsDiff(self.__image, backgroundI, differenceI) cv.CvtColor(differenceI, grayI, cv.CV_BGR2GRAY) cv.Smooth(grayI, grayI, cv.CV_BLUR, 5) cv.Smooth(grayI, grayI, cv.CV_MEDIAN, 5) cv.Threshold(grayI, grayI, threshold, 255, cv.CV_THRESH_BINARY) cv.SetZero(differenceI) cv.And(self.__image, self.__image, differenceI, grayI) return differenceI, grayI
def trackClean(self, objectColor, threshold, thickness, lastPoint2, imgScribble): """ Applies the technique of tracking to local image with the method of to clear. For this it is used the color of the followed object to show the lowed image.(ONLY WEBCAM) @param objectColor: Color of the used object to detect. @type objectColor: cvScalar @param threshold: Threshould detected color of the object. @type threshold: int [0,255] @param thickness: Thickness used to draw. @type thickness: int [0,255] @param lastPoint2: Auxiliar Object @param imgScribble: Image where is stored the tracking way. @type imgScribble: IplImage """ lastPoint = lastPoint2[0] imgThresh = self.__getThresholdImage1(threshold, objectColor) moments = cv.Moments(imgThresh, 1) moment10 = cv.GetSpatialMoment(moments, 1, 0) moment01 = cv.GetSpatialMoment(moments, 0, 1) area = cv.GetCentralMoment(moments, 0, 0) #print area if area == 0: currentPoint = None else: currentPoint = [moment10 / area, moment01 / area] if lastPoint != None and currentPoint != None: cv.Line(imgScribble, (int(currentPoint[0]), int(currentPoint[1])), (int(lastPoint[0]), int(lastPoint[1])), cv.Scalar(250, 250, 250), thickness, cv.CV_AA) mask = cv.CreateImage(cv.GetSize(self.__image), 8, 1) cv.SetZero(mask) cv.CvtColor(imgScribble, mask, cv.CV_BGR2GRAY) blackI = cv.CreateImage(cv.GetSize(self.__image), 8, 3) cv.SetZero(blackI) cv.And(self.__image, self.__image, blackI, mask) self.__image = blackI lastPoint2[0] = currentPoint
def __getBackGround1(self, threshold, color): """ Method number 2 to detect the background pixels. The background is detected of the local Image using given color. For this it is calculed the difference between local Image and another image set up the given color only in each one channel of color. @param threshold: Edge to detect the color background @type threshold: int @param color: Color used to detect background @type color: cvScalar @return: As first parameter the original image changing the different pixel to black pixel. As Second parameter a IplImage in Gray Scale where the differences are white pixels. After this can be used as mask. """ backgroundI = cv.CreateImage(cv.GetSize(self.__image), cv.IPL_DEPTH_8U, 3) cv.Set(backgroundI, color) i0, i1, i2, i3, b0, b1, b2, b3, d0, d1, d2, d3 = tuple([ cv.CreateImage(cv.GetSize(self.__image), cv.IPL_DEPTH_8U, 1) for i in range(12) ]) differenceI = cv.CreateImage(cv.GetSize(self.__image), cv.IPL_DEPTH_8U, 3) grayI = cv.CreateImage(cv.GetSize(differenceI), cv.IPL_DEPTH_8U, 1) cv.Split(self.__image, i0, i1, i2, None) cv.Split(backgroundI, b0, b1, b2, None) cv.AbsDiff(i0, b0, d0) cv.AbsDiff(i1, b1, d1) cv.AbsDiff(i2, b2, d2) cv.Threshold(d0, d0, threshold, 255, cv.CV_THRESH_BINARY) cv.Threshold(d1, d1, threshold, 255, cv.CV_THRESH_BINARY) cv.Threshold(d2, d2, threshold, 255, cv.CV_THRESH_BINARY) cv.Merge(d0, d1, d2, None, differenceI) cv.CvtColor(differenceI, grayI, cv.CV_BGR2GRAY) cv.Smooth(grayI, grayI, cv.CV_BLUR, 5) cv.Smooth(grayI, grayI, cv.CV_MEDIAN, 5) cv.Threshold(grayI, grayI, threshold, 255, cv.CV_THRESH_BINARY) cv.SetZero(differenceI) cv.Add(self.__image, differenceI, differenceI, grayI) return differenceI, grayI
def trackColor(self, objectColor, threshold, penColor, thickness, lastPoint2, imgScribble, typeS, typeDraw, disable): """ Applies the technique of tracking to local image. For this it is used the color of the followed object to draw the track in the image. It is possible apply two differents techniques to follow the object and also thickness is chosen. (ONLY WEBCAM) @param objectColor: Color of the used object to detect. @type objectColor: cvScalar @param threshold: Threshould detected color of the object. @type threshold: int [0,255] @param penColor: Drawing used color @type penColor: cvScalar @param thickness: Thickness used to draw. @type thickness: int [0,255] @param lastPoint2: Auxiliar Object @param imgScribble: Image where is stored the tracking way. @type imgScribble: IplImage @param typeS: Type of method used to detect the object @type typeS: int[0,1] @param typeDraw: If 1 the transparence is disabled. @param disable: Parameter of pause option """ lastPoint = lastPoint2[0] #penColor = objectColor imgThresh = None if typeS == 1: lowerColor = cv.Scalar(objectColor[0] - threshold, objectColor[1] - threshold, objectColor[2] - threshold) upperColor = cv.Scalar(objectColor[0] + threshold, objectColor[1] + threshold, objectColor[2] + threshold) imgThresh = self.__getThresholdImage(lowerColor, upperColor) elif typeS == 0: imgThresh = self.__getThresholdImage1(threshold, objectColor) moments = cv.Moments(imgThresh, 1) moment10 = cv.GetSpatialMoment(moments, 1, 0) moment01 = cv.GetSpatialMoment(moments, 0, 1) area = cv.GetCentralMoment(moments, 0, 0) #print area if area == 0 or disable == True: currentPoint = None else: currentPoint = [moment10 / area, moment01 / area] if lastPoint != None and currentPoint != None: cv.Line(imgScribble, (int(currentPoint[0]), int(currentPoint[1])), (int(lastPoint[0]), int(lastPoint[1])), penColor, thickness, cv.CV_AA) mask = cv.CreateImage(cv.GetSize(self.__image), 8, 1) cv.CvtColor(imgScribble, mask, cv.CV_BGR2GRAY) cv.Threshold(mask, mask, 20, 255, cv.CV_THRESH_BINARY) cv.Not(mask, mask) blackI = cv.CreateImage(cv.GetSize(self.__image), 8, 3) cv.SetZero(blackI) cv.And(self.__image, self.__image, blackI, mask) if typeDraw == 0: # Totally Transparent cv.Add(self.__image, imgScribble, self.__image) else: # Totally Opaque cv.Add(blackI, imgScribble, self.__image) #self.__image = blackI lastPoint2[0] = currentPoint