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

Formatting changes

Browse Source
This commit is contained in:
Vighnesh Birodkar
2015-06-11 00:20:09 +05:30
parent 1e49f49846
commit 06220903a1
2 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
doc/examples/plot_seam_carving.py
+3 -2
View File
@@ -66,12 +66,13 @@ importance of those pixels
masked_img = img.copy()
poly = [(404, 281), (404, 360), (359, 364), (338, 337), (145, 337), (120, 322), (145, 304), (340, 306), (362, 284)]
poly = [(404, 281), (404, 360), (359, 364), (338, 337), (145, 337), (120, 322),
(145, 304), (340, 306), (362, 284)]
pr = np.array([p[0] for p in poly])
pc = np.array([p[1] for p in poly])
rr, cc = draw.polygon(pr, pc)
masked_img[rr,cc,:] = masked_img[rr,cc,:]*0.5 + hl_color*.5
masked_img[rr, cc, :] = masked_img[rr, cc, :]*0.5 + hl_color*.5
plt.figure()
plt.title('Object Marked')
skimage/transform/_seam_carving.pyx
+8 -5
View File
@@ -41,14 +41,14 @@ cdef _preprocess_image(cnp.double_t[:, :, ::1] energy_img,
for offset in range(-1, 2):
c_idx = c + offset
if (c_idx > cols - 1) or (c_idx < 0) :
if (c_idx > cols - 1) or (c_idx < 0):
continue
if cumulative_img[r-1, c_idx] < min_cost:
min_cost = cumulative_img[r-1, c_idx]
track_img[r, c] = offset
cumulative_img[r,c] = min_cost + energy_img[r, c, 0]
cumulative_img[r, c] = min_cost + energy_img[r, c, 0]
cdef cnp.uint8_t _mark_seam(cnp.int8_t[:, ::1] track_img,
Py_ssize_t start_index,
@@ -98,7 +98,7 @@ cdef cnp.uint8_t _mark_seam(cnp.int8_t[:, ::1] track_img,
return 1
cdef _remove_seam(cnp.double_t[:, :, ::1] img,
cnp.uint8_t[:, ::1] seam_map, Py_ssize_t cols):
cnp.uint8_t[:, ::1] seam_map, Py_ssize_t cols):
""" Removes marked seams from an image.
Parameters
@@ -121,6 +121,7 @@ cdef _remove_seam(cnp.double_t[:, :, ::1] img,
for ch in range(channels):
img[r, c, ch] = img[r, c + shift, ch]
def _seam_carve_v(img, energy_map, iters, border):
""" Carve vertical seams off an image.
@@ -160,7 +161,8 @@ def _seam_carve_v(img, energy_map, iters, border):
cdef cnp.double_t[::1] last_row = last_row_obj
cdef Py_ssize_t[::1] sorted_indices
cdef cnp.uint8_t[:, ::1] seam_map = np.zeros(img.shape[0:2], dtype=np.uint8)
cdef cnp.uint8_t[:, ::1] seam_map = np.zeros(img.shape[0:2],
dtype=np.uint8)
cdef Py_ssize_t cols = img.shape[1]
cdef Py_ssize_t rows = img.shape[0]
cdef Py_ssize_t seams_left = iters
@@ -169,7 +171,8 @@ def _seam_carve_v(img, energy_map, iters, border):
cdef cnp.double_t[:, :, ::1] image = img
cdef cnp.int8_t[:, ::1] track_img = np.zeros(img.shape[0:2], dtype=np.int8)
cdef cnp.double_t[:, ::1] cumulative_img = np.zeros(img.shape[0:2], dtype=np.float)
cdef cnp.double_t[:, ::1] cumulative_img = np.zeros(img.shape[0:2],
dtype=np.float)
cdef cnp.double_t[:, :, ::1] energy_img
energy_map[:, 0:border] = DBL_MAX