zhigangjiang commited on
Commit
c4adcb2
1 Parent(s): 6ab9403

Update utils/boundary.py

Browse files
Files changed (1) hide show
  1. 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.float)
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