Spaces:
Build error
Build error
Update README.md
Browse files
README.md
CHANGED
|
@@ -11,42 +11,50 @@ pinned: false
|
|
| 11 |
|
| 12 |
# Segment Anything WebUI
|
| 13 |
|
| 14 |
-
[
|
| 15 |
|
| 16 |
-
|
| 17 |
|
| 18 |

|
| 19 |
|
| 20 |
-
##
|
| 21 |
|
| 22 |
-
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
|
| 24 |
```
|
| 25 |
pip install git+https://github.com/facebookresearch/segment-anything.git
|
| 26 |
```
|
| 27 |
|
| 28 |
-
- `git clone`
|
| 29 |
|
| 30 |
```
|
| 31 |
git clone https://github.com/5663015/segment_anything_webui.git
|
| 32 |
```
|
| 33 |
|
| 34 |
-
-
|
| 35 |
|
| 36 |
-
-
|
| 37 |
|
| 38 |
-
- `vit_l`: [ViT-L SAM model
|
| 39 |
|
| 40 |
-
- `vit_b`: [ViT-B SAM model
|
| 41 |
|
| 42 |
-
-
|
| 43 |
|
| 44 |
```
|
| 45 |
python app.py
|
| 46 |
```
|
| 47 |
|
| 48 |
-
|
| 49 |
|
| 50 |
## TODO
|
| 51 |
|
| 52 |
-
-
|
|
|
|
|
|
|
|
|
| 11 |
|
| 12 |
# Segment Anything WebUI
|
| 13 |
|
| 14 |
+
This project is based on **[Segment Anything Model](https://segment-anything.com/) ** by Meta. The UI is based on [Gradio](https://gradio.app/).
|
| 15 |
|
| 16 |
+
- Try deme on HF: [AIBoy1993/segment_anything_webui](https://huggingface.co/spaces/AIBoy1993/segment_anything_webui)
|
| 17 |
|
| 18 |

|
| 19 |
|
| 20 |
+
## Change Logs
|
| 21 |
|
| 22 |
+
- [2023-4-11] Support video segmentation.
|
| 23 |
+
|
| 24 |
+
## **Usage**
|
| 25 |
+
|
| 26 |
+
Following usage is running on your computer.
|
| 27 |
+
|
| 28 |
+
- Install Segment Anything([more details about install Segment Anything](https://github.com/facebookresearch/segment-anything#installation)):
|
| 29 |
|
| 30 |
```
|
| 31 |
pip install git+https://github.com/facebookresearch/segment-anything.git
|
| 32 |
```
|
| 33 |
|
| 34 |
+
- `git clone` this repository:
|
| 35 |
|
| 36 |
```
|
| 37 |
git clone https://github.com/5663015/segment_anything_webui.git
|
| 38 |
```
|
| 39 |
|
| 40 |
+
- Make a new folder named `checkpoints` under this project,and put the downloaded weights files in `checkpoints`。You can download the weights using following URLs:
|
| 41 |
|
| 42 |
+
- `vit_h`: [ViT-H SAM model](https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth)
|
| 43 |
|
| 44 |
+
- `vit_l`: [ViT-L SAM model](https://dl.fbaipublicfiles.com/segment_anything/sam_vit_l_0b3195.pth)
|
| 45 |
|
| 46 |
+
- `vit_b`: [ViT-B SAM model](https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth)
|
| 47 |
|
| 48 |
+
- Run:
|
| 49 |
|
| 50 |
```
|
| 51 |
python app.py
|
| 52 |
```
|
| 53 |
|
| 54 |
+
**Note:** Default model is `vit_b`,the demo can run on CPU. Default device is `cuda`。
|
| 55 |
|
| 56 |
## TODO
|
| 57 |
|
| 58 |
+
- Add segmentation prompt (point and box)
|
| 59 |
+
|
| 60 |
+
- Add text prompt
|