Spaces:
Sleeping
Sleeping
Commit
·
872f205
1
Parent(s):
be6042a
bg removal removed
Browse files
main.py
CHANGED
@@ -68,12 +68,12 @@ def get_masks():
|
|
68 |
|
69 |
masks = sorted(masks, key=(lambda x: x['area']), reverse=True)
|
70 |
|
71 |
-
def is_background(segmentation):
|
72 |
-
|
73 |
-
|
74 |
-
|
75 |
|
76 |
-
masks = [mask for mask in masks if not is_background(mask['segmentation'])]
|
77 |
|
78 |
for i in range(0, len(masks) - 1)[::-1]:
|
79 |
large_mask = masks[i]['segmentation']
|
|
|
68 |
|
69 |
masks = sorted(masks, key=(lambda x: x['area']), reverse=True)
|
70 |
|
71 |
+
# def is_background(segmentation):
|
72 |
+
# val = (segmentation[10, 10] or segmentation[-10, 10] or
|
73 |
+
# segmentation[10, -10] or segmentation[-10, -10])
|
74 |
+
# return val
|
75 |
|
76 |
+
# masks = [mask for mask in masks if not is_background(mask['segmentation'])]
|
77 |
|
78 |
for i in range(0, len(masks) - 1)[::-1]:
|
79 |
large_mask = masks[i]['segmentation']
|