Skip to content

Commit 160599d

Browse files
committed
fix mask bug
1 parent 4a144d4 commit 160599d

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

data_viewer/util.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -158,6 +158,7 @@ def calc_radial_profile(image, center, binsize=1., mask=None, mode='sum'):
158158
mask = (mask > 0.5).astype(np.float64)
159159
else:
160160
mask = np.ones_like(image)
161+
image *= mask # apply mask on input image
161162
y, x = np.indices((image.shape))
162163
r = np.sqrt((x - center[0])**2. + (y - center[1])**2.)
163164
bin_r = r / binsize
@@ -223,7 +224,7 @@ def calc_angular_profile(image, center, binsize=1., mask=None, mode='sum'):
223224
mask = (mask > 0.5).astype(np.float64)
224225
else:
225226
mask = np.ones_like(image)
226-
image = image * mask
227+
image *= mask
227228
y, x = np.indices((image.shape))
228229
theta = np.rad2deg(np.arctan2(y-center[1], x-center[0]))
229230
bin_theta = theta.copy()
@@ -281,7 +282,7 @@ def calc_across_center_line_profile(image, center, angle=0., width=1, mask=None,
281282
mask = (mask > 0.5).astype(np.float64)
282283
else:
283284
mask = np.ones_like(image)
284-
image = image * mask
285+
image *= mask
285286
# generate a larger image if the given center is not the center of the image.
286287
sy, sx = image.shape
287288
if sy % 2 == 0:

0 commit comments

Comments
 (0)