uint8 makss

app.py

@@ -35,8 +35,7 @@ with block:
     with gr.Column():
         gr.Markdown(
             '''# Segment Anything!🚀
-            The Segment Anything Model (SAM) produces high quality object masks from input prompts such as points or boxes, and it can be used to generate masks for all objects in an image. More information can be found in [**Official Project**](https://segment-anything.com/).
-            [![Duplicate this Space](https://huggingface.co/datasets/huggingface/badges/raw/main/duplicate-this-space-sm.svg)](https://huggingface.co/spaces/AIBoy1993/segment_anything_webui?duplicate=true)
+            The Segment Anything Model (SAM) produces high quality object masks from input prompts such as points or boxes, and it can be used to generate masks for all objects in an image. [**Official Project**](https://segment-anything.com/) [**Code**](https://github.com/facebookresearch/segment-anything).
             '''
         )
         with gr.Row():

inference.py

@@ -17,8 +17,8 @@ def PCL(mask, depth):
     assert mask.shape == depth.shape
     assert type(mask) == np.ndarray
     assert type(depth) == np.ndarray
-    rgb_mask = np.zeros((mask.shape[0], mask.shape[1], 3))
-    rgb_mask[mask] = (1.0, 0.0, 0.0)
+    rgb_mask = np.zeros((mask.shape[0], mask.shape[1], 3)).astype("uint8")
+    rgb_mask[mask] = (255, 0, 0)
     depth_o3d = o3d.geometry.Image(depth)
     image_o3d = o3d.geometry.Image(mask)
     rgbd_image = o3d.geometry.RGBDImage.create_from_color_and_depth(image_o3d, depth_o3d, convert_rgb_to_intensity=False)