Home

Awesome

SegDrawer

Simple static web-based mask drawer, supporting semantic drawing with interactive Segment Anything Model (SAM) and Video Segmentation Propagation with XMem, and with the stat-of-the-art Segment Anything Model 2 (SAM2).

<table> <tr> <td align="center"> <img src="example/demo.gif" width="240" /> </td> <td align="center"> <img src="example/demo1.gif" width="240" /> </td> <td align="center"> <img src="example/demo2.gif" width="240" /> </td> </tr> </table> <table> <tr> <td align="center"> <a href="https://www.youtube.com/watch?v=fn3KWM1kuAw">original video</a><br> <img src="example/video_seg/BostonDynamics_ori.gif" width="300" /> </td> <td align="center"> first frame seg<br> <img src="example/video_seg/BostonDynamics_mask.png" width="300" /> </td> </tr> <tr> <td align="center"> XMem<br> <img src="example/video_seg/BostonDynamics_seg.gif" width="300" /> </td> <td align="center"> SAM2<br> <img src="example/video_seg/BostonDynamics_sam2_seg.gif" width="300" /> </td> </tr> </table> <table> <tr> <td align="center"> Interactive segmentation<br> <img src="example/interactive_seg/interactive.gif" width="240" /> </td> <td align="center"> Revert during interactive seg.<br> <img src="example/interactive_seg/revert.gif" width="240" /> </td> </tr> </table>

Tools

From top to bottom

After Seg-Everything, the downloaded files would include .zip file, which contains all cut-offs. All these tools use SAM2 under the hood in default.

<table> <tr> <td align="center"> <img src="example/dog.jpg" width="360" /> </td> <td align="center"> <img src="example/cut-off.jpg" width="360" /> </td> </tr> </table>

For Video Segmentation with SAM2 or XMem, an initial segmentation map is needed, which can be easily achieved with SAM or SAM2. You can upload a video just as uploading an image, then draw a segmentation on it, after which you can click the final button of VideoSeg to upload it to the server and wait for the automatic download of video seg result.

Note: you may not want to draw the segmentation map manually with the tool Drawer (Same problem holds for Eraser), which leads to non-single color paints especially on the edge as shown below. This is not good for XMem video segmentation. For more details please refer to the original paper. Using SAM for segmenting is preferable.

<table> <tr> <td align="center"> <img src="example/drawer.png" width="360" /> </td> </tr> </table>

For Interactive Segmentation

  1. How to start
    • Click magic wand button (the curso becomes cross)
  2. How to use
    • Postive prompt by single left click
    • Negative prompt by single right click
    • The behavior of revert button will change, which removes the latest interactive prompt
  3. How to end
    • Click the magic wand button once again (the curso becomes normal). Note: it's actually safe to click any other button while in interactive mode.
    • The latest mask will save to the mask collections
    • The behavior of revert button will be turned back

Run Locally

If you don't need SAM for segmentation, just open segDrawer.html and use tools except SAM segmenter.

If you use SAM segmenter, do following steps (CPU can be time-consuming)

wget https://dl.fbaipublicfiles.com/segment_anything_2/072824/sam2_hiera_tiny.pt

or

wget https://dl.fbaipublicfiles.com/segment_anything/sam_vit_l_0b3195.pth
wget -P ./XMem/saves/ https://github.com/hkchengrex/XMem/releases/download/v1.0/XMem.pth
pip install -r requirements.txt
python server.py
http://127.0.0.1:7860

For configuring CPU/GPU and model, just change the code in server.py

sam_checkpoint = "sam_vit_l_0b3195.pth" # "sam_vit_l_0b3195.pth" or "sam_vit_h_4b8939.pth"
model_type = "vit_l" # "vit_l" or "vit_h"
device = "cuda" # "cuda" if torch.cuda.is_available() else "cpu"

Colab Tutorial

Follow this Colab example, or run on Colab. Need to register a ngrok account and copy your token to replace "{your_token}".

Development

There might be some bugs because this is an old version long ago and I haven't had the opportunity to fix them. Feel free to contribute by pulling requests or submitting issues.