Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -71,7 +71,7 @@ def get_seg(image, object):
|
|
71 |
mask_array_three_channel_invert = 1-mask_array_three_channel
|
72 |
#blur_image_reverse_mask = blur_image*mask_array_three_channel_invert
|
73 |
seg_box=[]
|
74 |
-
for i in range(pred):
|
75 |
object_number = int(object.split('i')[0])
|
76 |
mask_array = np.asarray(pred[object_number]['mask'])/255
|
77 |
image_array = np.asarray(image)
|
|
|
71 |
mask_array_three_channel_invert = 1-mask_array_three_channel
|
72 |
#blur_image_reverse_mask = blur_image*mask_array_three_channel_invert
|
73 |
seg_box=[]
|
74 |
+
for i in range(len(pred)):
|
75 |
object_number = int(object.split('i')[0])
|
76 |
mask_array = np.asarray(pred[object_number]['mask'])/255
|
77 |
image_array = np.asarray(image)
|