mirror image and reverse sort

app.py

@@ -2,7 +2,7 @@ import os
 import gradio as gr
 import numpy as np
 import cv2 
-from PIL import Image
+from PIL import Image, ImageOps
 import torch
 from inference import SegmentPredictor, DepthPredictor
 from utils import generate_PCL, PCL3, point_cloud
@@ -76,10 +76,11 @@ with block:
                   sam_decode_btn, depth_reconstruction_btn, prompt_image, lbl_image, n_samples, cube_size, selected_masks_image}
     
     def on_upload_image(input_image, upload_image):
-        return [upload_image, upload_image]
+        ## Mirror because gradio.image webcam has mirror = True
+        upload_image_mirror  = ImageOps.mirror(upload_image)
+        return [upload_image_mirror, upload_image]
     upload_image.upload(on_upload_image, [input_image, upload_image], [input_image, upload_image])
 
-
     # event - init coords
     def on_reset_btn_click(input_image):
         return input_image, point_coords_empty(), point_labels_empty(), None, []

inference.py

@@ -189,7 +189,7 @@ class SegmentPredictor:
             point_labels=lbls,
             multimask_output=True
             )
-        idxs = np.argsort(masks.sum(axis=(1,2)))
+        idxs = np.argsort(-masks.sum(axis=(1,2)))
         sam_masks = []
         for n,i in enumerate(idxs):
             sam_masks.append((masks[i], str(n)))