Python feature.match_template函数代码示例

def matchSections():
    pitches = cPickle.load(open('AmuInstPitches.pkl'))
    sections = cPickle.load(open('AmuInstSections.pkl'))
    filenames = cPickle.load(open('AmuFilenames.pkl'))
    keys = cPickle.load(open('AmuKeys.pkl'))
    for i in range(len(pitches)/2):
        pyplot.figure(i,(16,9))
        newp = []
        for vector in pitches[2*i]:
            deq = deque(vector)
            deq.rotate(keys[2*i+1]-keys[2*i])
            l = list(deq)
            newp.append(l)
        pitches[2*i] = newp
        image = numpy.array(pitches[2*i])
        template = numpy.array(pitches[2*i+1][sections[2*i+1][0]:sections[2*i+1][1]])
        im = feature.match_template(image,template,pad_input=True)
        pyplot.vlines(12,0,im.shape[0],'b')
        for j in range(len(sections[2*i+1])-2):    
            template = numpy.array(pitches[2*i+1][sections[2*i+1][j+1]:sections[2*i+1][j+2]])
            temp = feature.match_template(image,template,pad_input=True)
            im = numpy.concatenate((im,temp),axis = 1)
            pyplot.vlines(12*j+12,0,im.shape[0],'b')
        ij = numpy.unravel_index(numpy.argmax(im), im.shape)
        x, y = ij[::-1]
        pyplot.imshow(im, cmap = pyplot.get_cmap('gray'), aspect = 'auto', origin = 'lower')
        pyplot.ylabel(os.path.basename(filenames[2*i]) + " (in beats)")
        pyplot.xlabel(os.path.basename(filenames[2*i+1]) + " (12 Chroma Values Each)")
        pyplot.title('Section Similarity')
        pyplot.plot(x,y,'o',markeredgecolor='r',markerfacecolor='none',markersize=10)
        pyplot.xlim(0,im.shape[1]-1)
        pyplot.ylim(0,im.shape[0])
    pyplot.show()
    sys.exit()

def match_digit(image,templates,min_w=24,is_two_digits=True,debug=False):
    h,w = image.shape
    
    if is_two_digits:
        #first digit should be: 1,2,3,4,5
        ret = []
        
        wlimit = max(w/2,min_w)
        for digit in [1,2,3,4,5]:
            result = match_template(image[:,:wlimit], templates[digit])
            max_correl = np.max(result)
            ret += [(max_correl,digit)]
            if debug: print "digit1:",digit,max_correl
        #second digit should be any except if first digit is 5. That case options are 0,1,2,3,4
        if len(ret) > 0:
            ret.sort()
            #best digit is the last
            correl1,digit1 = ret[-1]
        else:
            return None,None

        digits = range(0,10) if digit1 != 5 else [0,1,2,3,4]
        ret = []
        for digit in digits:
            result = match_template(image[:,w - wlimit:], templates[digit])
            max_correl = np.max(result)
            ret += [(max_correl,digit)]
            if debug: print "digit2:",digit,max_correl
        if len(ret) > 0:
            ret.sort()
            #best digit is the last
            correl2,digit2 = ret[-1]
        else:
            return None,None
            
        return (correl1,digit1),(correl2,digit2)

    else:
        #assert h >= 20, "problem with w {0}:{1}".format(h,20)
        #assert w >= 18, "problem with h {0}:{1}".format(w,18)
        
        ret = []
        #one digit: 6,7,8,9
        for digit in [6,7,8,9]:
            result = match_template(image, templates[digit])
            max_correl = np.max(result)
            if debug: print "digit1:",digit,max_correl
            ret += [(max_correl,digit)]
        
        if len(ret) > 0:
            ret.sort()
            #best digit is the last
            return ret[-1],None

def find_bulb(image, templ):
    """finds the terminal bulb in an image using template correlation.

    Finds the best location (shifted cross-correlation) between image and template 
    return: location (x,y) and correlation value at the maximal correlation.

    """
    image = ndimage.gaussian_filter(image, 2) #- ndimage.gaussian_filter(res, 50)
    cut = int(0.1*image.shape[1])
    
    result = match_template(image, templ)
    xm = int(result.shape[1]/2.)
    res = result[:,max(0,-cut + xm):xm+cut]
    
    
    ij = np.unravel_index(np.argmax(res), res.shape)
    x0, y0 = ij[::-1]
    # calculate half template size
    t_half = int(templ.shape[0]/2.)
    conf = res[y0,x0]
    
    result1 = match_template(image, templ[t_half:,])
    res1 = result1[:,max(0,-cut + xm):xm+cut]
    ij = np.unravel_index(np.argmax(res1), res1.shape)
    
    x1, y1 = ij[::-1]
    conf1 = res1[y1,x1]
    if conf1 > conf:
        conf = conf1
        x0,y0 = x1,y1
        res = res1
        t_half = int(templ.shape[0]/4.)
            
    result2 = match_template(image, templ[:t_half,])
    res2 = result2[:,max(0,-cut + xm):xm+cut]

    ij = np.unravel_index(np.argmax(res2), res2.shape)
    x2, y2 = ij[::-1]
    conf2 = res2[y2,x2]
    if conf2 > conf:
        conf = conf2
        x0,y0 = x2,y2
        res = res2
        t_half = int(templ.shape[0])/4.
            
    x = max(0, min(x0+templ.shape[1]/2.+cut, image.shape[1]-1))
    y = max(0,min(y0+t_half, image.shape[0]-1))
    if conf < 0.4 or conf/np.std(res) < 2.5:
        conf = 0.0 
    return y,x, conf

def test_normalization():
    """Test that `match_template` gives the correct normalization.

    Normalization gives 1 for a perfect match and -1 for an inverted-match.
    This test adds positive and negative squares to a zero-array and matches
    the array with a positive template.
    """
    n = 5
    N = 20
    ipos, jpos = (2, 3)
    ineg, jneg = (12, 11)
    image = np.full((N, N), 0.5)
    image[ipos:ipos + n, jpos:jpos + n] = 1
    image[ineg:ineg + n, jneg:jneg + n] = 0

    # white square with a black border
    template = np.zeros((n + 2, n + 2))
    template[1:1 + n, 1:1 + n] = 1

    result = match_template(image, template)

    # get the max and min results.
    sorted_result = np.argsort(result.flat)
    iflat_min = sorted_result[0]
    iflat_max = sorted_result[-1]
    min_result = np.unravel_index(iflat_min, result.shape)
    max_result = np.unravel_index(iflat_max, result.shape)

    # shift result by 1 because of template border
    assert np.all((np.array(min_result) + 1) == (ineg, jneg))
    assert np.all((np.array(max_result) + 1) == (ipos, jpos))

    assert np.allclose(result.flat[iflat_min], -1)
    assert np.allclose(result.flat[iflat_max], 1)

def test_template():
    size = 100
    # Float prefactors ensure that image range is between 0 and 1
    image = np.full((400, 400), 0.5)
    target = 0.1 * (np.tri(size) + np.tri(size)[::-1])
    target_positions = [(50, 50), (200, 200)]
    for x, y in target_positions:
        image[x:x + size, y:y + size] = target
    np.random.seed(1)
    image += 0.1 * np.random.uniform(size=(400, 400))

    result = match_template(image, target)
    delta = 5

    positions = peak_local_max(result, min_distance=delta)

    if len(positions) > 2:
        # Keep the two maximum peaks.
        intensities = result[tuple(positions.T)]
        i_maxsort = np.argsort(intensities)[::-1]
        positions = positions[i_maxsort][:2]

    # Sort so that order matches `target_positions`.
    positions = positions[np.argsort(positions[:, 0])]

    for xy_target, xy in zip(target_positions, positions):
        assert_almost_equal(xy, xy_target)

def processImages():
    sims = cPickle.load(open('AmuInstSimMats.pkl'))
    for i,sim in enumerate(sims):
        pyplot.figure(0,(16,9))
        pyplot.imshow(sim, vmin = 0, vmax = 1, cmap = pyplot.get_cmap('gray'), aspect = 'auto', origin = 'lower')
        pyplot.title('Unfiltered Sim Matrix ' + str(i))
        pyplot.savefig('Unfiltered Sim Matrix ' + str(i) + '.jpg')
        pyplot.figure(1,(16,9))
        pyplot.imshow(filter.tv_denoise(numpy.array(sim,numpy.float64), weight = 1), vmin = 0, vmax = 1, cmap = pyplot.get_cmap('gray'), aspect = 'auto', origin = 'lower')
        pyplot.title('TV_Denoise ' + str(i))
        pyplot.savefig('TV_Denoise ' + str(i) + '.jpg')
        pyplot.figure(2,(16,9))
        pyplot.imshow(filter.threshold_adaptive(numpy.array(sim,numpy.float64),21), vmin = 0, vmax = 1, cmap = pyplot.get_cmap('gray'), aspect = 'auto', origin = 'lower')
        pyplot.title('Threshold_Adaptive ' + str(i))
        pyplot.savefig('Threshold_Adaptive ' + str(i) + '.jpg')
        pyplot.figure(3,(16,9))
        pyplot.imshow(ndimage.minimum_filter(numpy.array(sim,numpy.float64),size=2), vmin = 0, vmax = 1, cmap = pyplot.get_cmap('gray'), aspect = 'auto', origin = 'lower')
        pyplot.title('Local Minimum_Filter ' + str(i))
        pyplot.savefig('Local Minimum_Filter ' + str(i) + '.jpg')
        pyplot.figure(4,(16,9))
        template = numpy.array([[0,1,1,1,1,1,1,1],[1,0,1,1,1,1,1,1],[1,1,0,1,1,1,1,1],[1,1,1,0,1,1,1,1],
                                [1,1,1,1,0,1,1,1],[1,1,1,1,1,0,1,1],[1,1,1,1,1,1,0,1],[1,1,1,1,1,1,1,0]])
        pyplot.imshow(feature.match_template(numpy.array(sim,numpy.float64),template), vmin = 0, vmax = 1, cmap = pyplot.get_cmap('gray'), aspect = 'auto', origin = 'lower')
        pyplot.title('Match_Template with my own 8x8 beat diagonal template ' + str(i))
        pyplot.savefig('Match_Template with my own 8x8 beat diagonal template ' + str(i) + '.jpg')
    sys.exit()

def CalculateImageShift(imgref,img):
    imgH,imgW = imgref.shape
    result = match_template(imgref,img,pad_input=True)
    ij = np.unravel_index(np.argmax(result),result.shape)
    sx,sy = ij[::-1]
    sx,sy = imgW/2-sx,imgH/2-sy
    return sx,sy

    def transform(self, X):
        from skimage.feature import match_template
        X_out = None
        n_templates = len(self.template)
        raw = self.raw
        for i, x in enumerate(X):
            if i % 1000 == 0:
                print i

            for j, template in enumerate(self.template):
                result = match_template(x, template, pad_input=True)
                if X_out is None:
                    if raw:
                        dtype = (X.shape[0], n_templates, result.shape[0],
                                 result.shape[1])
                    else:
                        dtype = (X.shape[0], n_templates)
                    X_out = np.empty(dtype, dtype=np.float32)

                if not raw:
                    result = np.max(result)

                X_out[i, j] = result

        if raw:
            X_out = np.max(X_out, axis=1)
        return X_out

def test_pad_input():
    """Test `match_template` when `pad_input=True`.

    This test places two full templates (one with values lower than the image
    mean, the other higher) and two half templates, which are on the edges of
    the image. The two full templates should score the top (positive and
    negative) matches and the centers of the half templates should score 2nd.
    """
    # Float prefactors ensure that image range is between 0 and 1
    template = 0.5 * diamond(2)
    image = 0.5 * np.ones((9, 19))
    mid = slice(2, 7)
    image[mid, :3] -= template[:, -3:]  # half min template centered at 0
    image[mid, 4:9] += template         # full max template centered at 6
    image[mid, -9:-4] -= template       # full min template centered at 12
    image[mid, -3:] += template[:, :3]  # half max template centered at 18

    result = match_template(image, template, pad_input=True,
                            constant_values=image.mean())

    # get the max and min results.
    sorted_result = np.argsort(result.flat)
    i, j = np.unravel_index(sorted_result[:2], result.shape)
    assert_equal(j, (12, 0))
    i, j = np.unravel_index(sorted_result[-2:], result.shape)
    assert_equal(j, (18, 6))

def detect_start_end_times(pattern_wav, recording_wav, sr, overlap):
    """Find matches for the start/end pattern within the recorded audio"""

    # Compute the STFT of the recordings
    specgram1 = numpy.array(stft.spectrogram(pattern_wav, overlap=overlap))
    specgram2 = numpy.array(stft.spectrogram(recording_wav, overlap=overlap))

    # Restrict the spectrum to the frequency band occupied by the start/end pattern
    pattern = abs(specgram1[7:16,:])
    recording = abs(specgram2[7:16,:])

    # Search for matches of the pattern in the input recording and return a confidence score
    # for each time position of the input recording
    confidence = match_template(recording, pattern)

    # Search for peaks in the confidence score, and choose the two highest peaks
    # Minimum distance between consecutive peaks is set to 1 second
    peaks = peakutils.indexes(confidence[0], thres=0, min_dist=seconds_to_samples(1, overlap, sr))
    peaks = sorted(peaks, key=lambda p: -confidence[0,p])[:2]

    #TODO: throw errors instead of printing, if necessary
    if len(peaks) < 1:
        print "Could not detect a starting beep!"
    elif len(peaks) < 2:
        print "Could only detect one starting beep!"
    else:
        start, end = sorted(peaks)
        print "Initial beep detected at " + "%.3f" % samples_to_seconds(start, overlap, sr) + " seconds."
        print "Final beep detected at " + "%.3f" % samples_to_seconds(end, overlap, sr) + " seconds."
    return samples_to_seconds(start, overlap, sr), samples_to_seconds(end, overlap, sr)

	def run(self, im, skin_thresh=[-1,1], n_peaks=3):
		'''
		im : color image
		'''
		im_skin = im
		self.im_skin = im_skin
		skin_match_c = match_template(im_skin, self.template, pad_input=True)*(im>0)
		self.skin_match = skin_match_c
		# cv2.matchTemplate(im_skin, self.template, cv2.cv.CV_TM_SQDIFF_NORMED)
		# imshow(cv2.matchTemplate(im_skin.astype(np.float32), self.template.astype(np.float32), cv2.cv.CV_TM_CCOEFF_NORMED))

		# Display Predictions - Color Based matching
		optima = peak_local_max(skin_match_c, min_distance=20, num_peaks=n_peaks, exclude_border=False)
		# Visualize
		if len(optima) > 0:
			optima_values = skin_match_c[optima[:,0], optima[:,1]]
			optima_thresh = np.max(optima_values) / 2
			optima = optima.tolist()

			for i,o in enumerate(optima):
				if optima_values[i] < optima_thresh:
					optima.pop(i)
					break
		self.markers = optima

		return self.markers

def findTemplateInImage(templatePath, imagePath, debug=False):
	"""
	Returns the position of a template in an image
	"""
# load template
	template = skimage.transform.rescale(io.imread(templatePath), 0.5)
	if debug:
		plt.imshow(template)
		plt.title("Template")
		plt.show()

# load image 
	image = io.imread(imagePath, False);
	image = skimage.transform.rescale(image, 0.5)
	if debug:
		plt.imshow(image)
		plt.title("Image")
		plt.show()

# find building position
	result = match_template(image, template)
	result = result.squeeze()
	ij = numpy.unravel_index(numpy.argmax(result), result.shape)
	x, y = ij[::-1]

# add image midpoint
	x += int(float(len(template[:]))/2)
	y += int(float(len(template))/2)

# re-rescale ;)
	return numpy.array([x*2, y*2])

def align_converge(y_LR,size=64):
    """iterate until offsets converge"""
    (h,w) = y_LR.shape
    # split image
    y_L = y_LR[:,:w/2]
    y_R = y_LR[:,w/2:]
    (h,w) = y_L.shape
    s = size / 2
    # now find n offsets
    rand = RandomState(0)
    prev_dx, prev_dy = 0, 0
    series = []
    while True:
        # at a random locations in y_L
        y = rand.randint(h/4,h*3/4)
        x = rand.randint(w/4,w*3/4)
        it = y_L[y:y+s,x:x+s] # take an s x s chunk there
        tm = match_template(y_R,it) # match it against y_R
        ry, rx = maximum_position(tm) # max value is location
        series += [((y-ry), (x-rx))] # accumulatea
        print series
        n = len(series)
        if n % 2 == 0:
            # take the median
            dy, dx = np.median(np.asarray(series),axis=0).astype(int)
            if n > 100 or (abs(dy-prev_dy) == 0 and abs(dx-prev_dx) == 0):
                return dy, dx
            prev_dy, prev_dx = dy, dx

def test_bounding_values():
    image = img_as_float(data.page())
    template = np.zeros((3, 3))
    template[1, 1] = 1
    result = match_template(img_as_float(data.page()), template)
    print(result.max())
    assert result.max() < 1 + 1e-7
    assert result.min() > -1 - 1e-7

 def _template_matching_shift(self, im1, im2, template):
     index = []
     for im in [im1, im2]:
         match = match_template(im, template)
         index.append(np.unravel_index(np.argmax(match), match.shape))
     index = np.array(index)
     shift = index[1] - index[0]
     return shift