zhigangjiang
commited on
Commit
•
c4adcb2
1
Parent(s):
6ab9403
Update utils/boundary.py
Browse files- utils/boundary.py +1 -1
utils/boundary.py
CHANGED
@@ -273,7 +273,7 @@ def get_heat_map(u_s, patch_num=256, sigma=2, window_width=15, show=False):
|
|
273 |
gauss_map = get_gauss_map(sigma, window_width)
|
274 |
heat_map_all = []
|
275 |
for u in pixel_us:
|
276 |
-
heat_map = np.zeros(patch_num, dtype=np.
|
277 |
left = u-window_width
|
278 |
right = u+window_width+1
|
279 |
|
|
|
273 |
gauss_map = get_gauss_map(sigma, window_width)
|
274 |
heat_map_all = []
|
275 |
for u in pixel_us:
|
276 |
+
heat_map = np.zeros(patch_num, dtype=np.float32)
|
277 |
left = u-window_width
|
278 |
right = u+window_width+1
|
279 |
|