Spaces:
Runtime error
Runtime error
File size: 3,113 Bytes
3c06936 c694bbe 3c06936 b5e4671 d53184c 3c06936 3a03ccd 404fd96 3ad7b92 404fd96 3ad7b92 404fd96 3ad7b92 404fd96 3ad7b92 404fd96 54c5ed9 3c06936 404fd96 a3f3782 404fd96 b5e4671 404fd96 3c06936 fcd9c8d 404fd96 b5e4671 3c06936 b5e4671 3ad7b92 b5e4671 3ad7b92 b5e4671 c64775c 3c06936 3ad7b92 3c06936 3ad7b92 404fd96 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 |
# sam2-playground
Gradio based playground web UI using [facebook/segment-anything-2](https://github.com/facebookresearch/segment-anything-2) models.
### Online Demos
[](https://colab.research.google.com/github/jhj0517/sam2-playground/blob/master/notebooks/sam2_playground.ipynb)
[](https://huggingface.co/spaces/jhj0517/sam2-playground)
## Feature
- Add filters to segment parts of a video with pixelated or solid color
<table>
<tr>
<td align="center"><strong>Pixelize the girl's face</strong></td>
</tr>
<tr>
<td>
<video controls autoplay loop src="https://github.com/user-attachments/assets/c5758970-dc15-4bc8-a918-8d3e8e44a73a" muted="false"></video>
</td>
</tr>
</table>
- Divide segmentation parts into layers and save them as PSD files.
<table>
<tr>
<td align="center"><strong>Divide clothes into layers and save as PSD file</strong></td>
</tr>
<tr>
<td style="text-align: center;">
<img src="https://github.com/jhj0517/sam2-playground/blob/master/docs/example_psd_file.png" alt="Example_PSD">
</td>
</tr>
</table>
See [PROMPT_GUIDE.md](https://github.com/jhj0517/sam2-playground/blob/master/docs/PROMPT_GUIDE.md) to see how to prompt the segmentation parts of the image.
# Installation and Running
### Prerequisites
To run this Web UI, you need these prerequisites. If you don't have them, please install them in the following links :
- `git` : https://git-scm.com/downloads
- `python=>3.10` : https://www.python.org/downloads/
- `FFmpeg` : https://ffmpeg.org/download.html
After installing FFmpeg, **make sure to add the `FFmpeg/bin` folder to your system PATH!** <br>
And for CUDA, if you're not using an Nvidia GPU and CUDA 12.4, edit the [requirements.txt](https://github.com/jhj0517/sam2-playground/blob/master/requirements.txt) to match your environment.
### Option 1 : Running with Shell / Batch Scripts
There's a set of shell / batch scripts for installation and running.
1. Download [`sam2-playground-portable.zip`](https://github.com/jhj0517/sam2-playground/releases/tag/v1.0.0) with the file corresponding to your OS and extract its contents.
2. Run `install.bat` or `install.sh` to install dependencies. (This will create a `venv` directory and install dependencies there.)
3. Start WebUI with `start-webui.bat` or `start-webui.sh`
4. To update, run `update.bat` or `update.sh`
### Option 2: Docker
1. Clone the repository
```
git clone https://github.com/jhj0517/sam2-playground.git
```
3. Build the image ( Image is about ~6 GB )
```
docker compose build
```
3. Run the container
```
docker compose up
```
4. Connect to `localhost:7860` with your browser.
If needed, update [`docker-compose.yaml`](https://github.com/jhj0517/sam2-playground/blob/master/docker-compose.yaml) to match your environments.
## Todo 🗓
- [ ] Support `change()` API for `gradio_image_prompter` and automatically generate preview for video predictor
|