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

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -152,7 +152,7 @@ def show_alpha_floorplan(dt_xyz, side_l=512, border_color=None):
152
  dt_floorplan = draw_floorplan(xz=dt_xyz[..., ::2], fill_color=fill_color,
153
  border_color=border_color, side_l=side_l, show=False, center_color=[1, 0, 0, 1])
154
  dt_floorplan = Image.fromarray((dt_floorplan * 255).astype(np.uint8), mode='RGBA')
155
- back = np.zeros([side_l, side_l, len(fill_color)], dtype=np.float)
156
  back[..., :] = [0.8, 0.8, 0.8, 1]
157
  back = Image.fromarray((back * 255).astype(np.uint8), mode='RGBA')
158
  iou_floorplan = Image.alpha_composite(back, dt_floorplan).convert("RGB")
 
152
  dt_floorplan = draw_floorplan(xz=dt_xyz[..., ::2], fill_color=fill_color,
153
  border_color=border_color, side_l=side_l, show=False, center_color=[1, 0, 0, 1])
154
  dt_floorplan = Image.fromarray((dt_floorplan * 255).astype(np.uint8), mode='RGBA')
155
+ back = np.zeros([side_l, side_l, len(fill_color)], dtype=np.float32)
156
  back[..., :] = [0.8, 0.8, 0.8, 1]
157
  back = Image.fromarray((back * 255).astype(np.uint8), mode='RGBA')
158
  iou_floorplan = Image.alpha_composite(back, dt_floorplan).convert("RGB")