Update utils.py
Browse files
utils.py
CHANGED
|
@@ -18,6 +18,7 @@ def add_mask(image, label, vis_labels, colors, alpha=0.5):
|
|
| 18 |
|
| 19 |
|
| 20 |
def find_haight(mask):
|
|
|
|
| 21 |
v_sum = (np.sum(mask, axis=1) > 0).astype(np.uint8)
|
| 22 |
|
| 23 |
v_diff = np.diff(np.hstack((0, v_sum, 0)))
|
|
|
|
| 18 |
|
| 19 |
|
| 20 |
def find_haight(mask):
|
| 21 |
+
|
| 22 |
v_sum = (np.sum(mask, axis=1) > 0).astype(np.uint8)
|
| 23 |
|
| 24 |
v_diff = np.diff(np.hstack((0, v_sum, 0)))
|