Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -136,7 +136,7 @@ def floorplan_generation(outline, num_of_rooms):
|
|
| 136 |
image_arr = np.array(image)
|
| 137 |
mask_bgr = cv2.cvtColor(image_arr, cv2.COLOR_RGB2BGR)
|
| 138 |
mask_gray = cv2.cvtColor(mask_bgr, cv2.COLOR_RGB2GRAY)
|
| 139 |
-
mask_white = cv2.threshold(mask_gray,
|
| 140 |
image_arr[mask_white<250]=(255,255,255)
|
| 141 |
image_arr_copy = image_arr.copy()
|
| 142 |
|
|
|
|
| 136 |
image_arr = np.array(image)
|
| 137 |
mask_bgr = cv2.cvtColor(image_arr, cv2.COLOR_RGB2BGR)
|
| 138 |
mask_gray = cv2.cvtColor(mask_bgr, cv2.COLOR_RGB2GRAY)
|
| 139 |
+
mask_white = cv2.threshold(mask_gray, 200, 255, cv2.THRESH_BINARY_INV)[1]
|
| 140 |
image_arr[mask_white<250]=(255,255,255)
|
| 141 |
image_arr_copy = image_arr.copy()
|
| 142 |
|