AIBoy1993 commited on
Commit
9fbfe03
·
1 Parent(s): 2b7384b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -12
README.md CHANGED
@@ -11,42 +11,50 @@ pinned: false
11
 
12
  # Segment Anything WebUI
13
 
14
- [[`Paper`](https://ai.facebook.com/research/publications/segment-anything/)] [[`Project`](https://segment-anything.com/)] [[`Demo`](https://segment-anything.com/demo)] [[`Dataset`](https://segment-anything.com/dataset/index.html)] [[`Blog`](https://ai.facebook.com/blog/segment-anything-foundation-model-image-segmentation/)]
15
 
16
- 本项目是 Meta 最新成果 [Segment Anything Model](https://segment-anything.com/) 的简单 WebUI,基于 [Gradio](https://gradio.app/)。虽然官方有很好的 [demo](https://segment-anything.com/demo),但也算是通过这么个简单的 UI 熟悉一下这个成果的用法。目前只实现了自动分割。
17
 
18
  ![](./images/20230408023615.png)
19
 
20
- ## **用法:**
21
 
22
- - 安装 Segment Anything([其他安装方式](https://github.com/facebookresearch/segment-anything#installation)):
 
 
 
 
 
 
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
- - 在项目目录下新建 `checkpoints` 文件夹,将下载的模型文件放进去。模型文件下载:
35
 
36
- - **`default` or `vit_h`: [ViT-H SAM model.](https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth)**
37
 
38
- - `vit_l`: [ViT-L SAM model.](https://dl.fbaipublicfiles.com/segment_anything/sam_vit_l_0b3195.pth)
39
 
40
- - `vit_b`: [ViT-B SAM model.](https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth)
41
 
42
- - 运行:
43
 
44
  ```
45
  python app.py
46
  ```
47
 
48
- **注意:** 默认模型是 `vit_b`,笔记本 CPU 可运行。运行硬件默认是 `cuda`。
49
 
50
  ## TODO
51
 
52
- - 增加 segmentation prompt
 
 
 
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
  ![](./images/20230408023615.png)
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