diff --git a/.idea/ISAT_with_segment_anything.iml b/.idea/ISAT_with_segment_anything.iml index a34782b..e7438cb 100644 --- a/.idea/ISAT_with_segment_anything.iml +++ b/.idea/ISAT_with_segment_anything.iml @@ -2,7 +2,7 @@ - + \ No newline at end of file diff --git a/.idea/misc.xml b/.idea/misc.xml index b47cefa..3f589e5 100644 --- a/.idea/misc.xml +++ b/.idea/misc.xml @@ -1,4 +1,4 @@ - + \ No newline at end of file diff --git a/README-en.md b/README-en.md index 13d2efe..386fba9 100644 --- a/README-en.md +++ b/README-en.md @@ -1,47 +1,60 @@ # ISAT with segment anything # Image segmentation annotation tool with segment anything. +[标注.gif](./display/标注%20-big-original.gif) -Quick annotate image by [segment anything](https://github.com/facebookresearch/segment-anything). +Quick annotate for image segmentation by [segment anything](https://github.com/facebookresearch/segment-anything) + +**Project updating,suggestions welcome** + +Demo Video:[youtube](https://www.youtube.com/watch?v=yLdZCPmX-Bc) + +演示视频:[bilibili](https://www.bilibili.com/video/BV1Lk4y1J7uB/) [中文](README.md) [English](README-en.md) - ## INSTALL -### 1. 源码运行 +### 1. Run the source code +#### (1) Create environment ```shell -# 创建虚拟环境 conda create -n ISAT_with_segment_anything python==3.8 conda activate ISAT_with_segment_anything ``` +#### (2) Install Segment anything ```shell -# Install Segment anything git clone git@github.com:facebookresearch/segment-anything.git cd segment-anything pip install -e . cd .. ``` +#### (3) Install ISAT_with_segment_anything ```shell -# Install ISAT_with_segment_anything git clone https://github.com/yatengLG/ISAT_with_segment_anything.git cd ISAT_with_segment_anything pip install -r requirements.txt ``` -```text -# Download Segment anything pretrained checkpoint. +#### (4) Download Segment anything pretrained checkpoint. + Download the checkpoint,and save in the path: ISAT_with_segment_anything/segment_any -checkpoint link: - https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth - https://dl.fbaipublicfiles.com/segment_anything/sam_vit_l_0b3195.pth - https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth -The checkpoint named sam_vit_h_4b8939 has best effect, but need more resources; -The checkpoint named sam_vit_b_01ec64 has poor effect, but need less resources; +- H-checkpoint:[sam_vit_h_4b8939.pth](https://dl.fbaipublicfiles.com/segment_anything/sam_vit_h_4b8939.pth) + + H checkpoint has best effect, but need more resources.VRAM needs at least 8G. +- L-checkpoint:[sam_vit_l_0b3195.pth](https://dl.fbaipublicfiles.com/segment_anything/sam_vit_l_0b3195.pth) + + L checkpoint has normal effect and normal resources.VRAM needs at least 7G. +- B-checkpoint:[sam_vit_b_01ec64.pth](https://dl.fbaipublicfiles.com/segment_anything/sam_vit_b_01ec64.pth) + + B checkpoint has pool effect, but need less resources.VRAM needs at least 6G. + +#### (5) Run +```shell +python main.py ``` -```shell -# Run the software -python main.py -``` \ No newline at end of file +# Explain +1. The software needs to run well under the GPU and enought VRAM. +2. If you don't have a GPU or don't have enought VARM, please manually draw polygons for labeling by [ISAT](https://github.com/yatengLG/ISAT). +3. \ No newline at end of file diff --git a/README.md b/README.md index 02aaf39..279bef4 100644 --- a/README.md +++ b/README.md @@ -1,11 +1,14 @@ # ISAT with segment anything # ISAT 图像分割标注工具(集成segment anything) +[标注.gif](./display/标注%20-big-original.gif) 集成[segment anything](https://github.com/facebookresearch/segment-anything),实现图片分割快速标注。 **项目持续更新中,[更新日志](./UpdateLog.md),欢迎大家提出建议** -演示视频:https://www.bilibili.com/video/BV1Lk4y1J7uB/ +演示视频:[bilibili](https://www.bilibili.com/video/BV1Lk4y1J7uB/) + +Demo Video:[youtube](https://www.youtube.com/watch?v=yLdZCPmX-Bc) [中文](README.md) [English](README-en.md) ## 特点 diff --git a/display/双语界面 -big-original.gif b/display/双语界面 -big-original.gif new file mode 100644 index 0000000..3e5d2af Binary files /dev/null and b/display/双语界面 -big-original.gif differ diff --git a/display/标注 -big-original.gif b/display/标注 -big-original.gif new file mode 100644 index 0000000..17ce176 Binary files /dev/null and b/display/标注 -big-original.gif differ diff --git a/display/编辑 -big-original.gif b/display/编辑 -big-original.gif new file mode 100644 index 0000000..9767dab Binary files /dev/null and b/display/编辑 -big-original.gif differ diff --git a/display/遮挡 -big-original.gif b/display/遮挡 -big-original.gif new file mode 100644 index 0000000..58d9bef Binary files /dev/null and b/display/遮挡 -big-original.gif differ