wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-07-29 11:26:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

Refactor final slicing

Browse Source
This commit is contained in:
Johannes Schönberger
2013-12-10 09:19:47 +01:00
parent 598ee3f9ec
commit b815b5d045
1 changed files with 8 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/feature/template.py
+8 -21
View File
@@ -156,27 +156,14 @@ def match_template(image, template, pad_input=False, mode='constant',
response[mask] = nom[mask] / denom[mask]
if pad_input:
r0 = (template.shape[0] - 1) // 2
r1 = r0 + image_shape[0]
c0 = (template.shape[1] - 1) // 2
c1 = c0 + image_shape[1]
else:
r0 = template.shape[0] - 1
r1 = r0 + image_shape[0] - template.shape[0] + 1
c0 = template.shape[1] - 1
c1 = c0 + image_shape[1] - template.shape[1] + 1
if image.ndim == 3:
slices = []
for i in range(template.ndim):
if pad_input:
d0 = (template.shape[2] - 1) // 2
d1 = d0 + image_shape[2]
d0 = (template.shape[i] - 1) // 2
d1 = d0 + image_shape[i]
else:
d0 = template.shape[2] - 1
d1 = d0 + image_shape[2] - template.shape[2] + 1
d0 = template.shape[i] - 1
d1 = d0 + image_shape[i] - template.shape[i] + 1
slices.append(slice(d0, d1))
response = response[r0:r1, c0:c1, d0:d1]
else:
response = response[r0:r1, c0:c1]
return response
return response[slices]