This commit is contained in:
yatenglg 2023-04-25 22:55:47 +08:00
parent a3912fc643
commit 21d12fe031
8 changed files with 37 additions and 21 deletions

View File

@ -2,7 +2,7 @@
<module type="PYTHON_MODULE" version="4">
<component name="NewModuleRootManager">
<content url="file://$MODULE_DIR$" />
<orderEntry type="jdk" jdkName="Python 3.6 (segment_anything_env)" jdkType="Python SDK" />
<orderEntry type="jdk" jdkName="Python 3.8 (ISAT_with_segment_anything)" jdkType="Python SDK" />
<orderEntry type="sourceFolder" forTests="false" />
</component>
</module>

2
.idea/misc.xml generated
View File

@ -1,4 +1,4 @@
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="ProjectRootManager" version="2" project-jdk-name="Python 3.6 (segment_anything_env)" project-jdk-type="Python SDK" />
<component name="ProjectRootManager" version="2" project-jdk-name="Python 3.8 (ISAT_with_segment_anything)" project-jdk-type="Python SDK" />
</project>

View File

@ -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 updatingsuggestions 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 checkpointand 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
```
# 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.

View File

@ -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)
## 特点

Binary file not shown.

After

Width:  |  Height:  |  Size: 476 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 2.4 MiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 2.2 MiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 724 KiB