Update app.py

app.py

@@ -19,6 +19,7 @@ def img_resize(image):
 
 def image_objects(image):
     global pred
+    
     image = img_resize(image)
     pred = model(image)
     pred_object_list = [str(i)+'_'+x['label'] for i, x in enumerate(pred)]
@@ -56,24 +57,11 @@ def blurr_object(image, object, blur_strength):
 def get_seg(image, object):
     image = img_resize(image)
 
-    object_number = int(object.split('_')[0])
-    mask_array = np.asarray(pred[object_number]['mask'])/255
-    image_array = np.asarray(image)
-
-    mask_array_three_channel = np.zeros_like(image_array)
-    mask_array_three_channel[:,:,0] = mask_array
-    mask_array_three_channel[:,:,1] = mask_array
-    mask_array_three_channel[:,:,2] = mask_array
-
-    segmented_image = image_array*mask_array_three_channel
-
-    #blur_image = np.asarray(image.filter(ImageFilter.GaussianBlur(radius=blur_strength)))
-    mask_array_three_channel_invert = 1-mask_array_three_channel
     #blur_image_reverse_mask = blur_image*mask_array_three_channel_invert
     seg_box=[]
     for i in range(len(pred)):
-        object_number = int(object.split('i')[0])
-        mask_array = np.asarray(pred[object_number]['mask'])/255
+        object_number = int(object.split('_')[0])
+        mask_array = np.asarray(pred[i]['mask'])/255
         image_array = np.asarray(image)
     
         mask_array_three_channel = np.zeros_like(image_array)