zhigangjiang
commited on
Commit
•
6ab9403
1
Parent(s):
54dbfd8
Update inference.py
Browse files- 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.
|
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")
|