doublelotus commited on
Commit
872f205
·
1 Parent(s): be6042a

bg removal removed

Browse files
Files changed (1) hide show
  1. main.py +5 -5
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
- 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']
 
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']