项目简介
本项目名为"ISAT with segment anything",是基于PyQt5框架开发的交互式图像标注工具。它集成了"segment anything"模型,可实现图像分割的快速标注,支持语义分割和实例分割。具备交互式修正mask、手动创建多边形、二次修改、调整多边形遮挡关系、预览标注结果等功能。此外,还支持将ISAT格式json转换为VOC、COCO、LabelMe格式,以及从COCO格式转换为ISAT格式。
项目的主要特性和功能
- 集成"segment anything"模型,实现交互式半自动图像分割标注。
- 支持通过鼠标左(右)键点击感兴趣(不感兴趣)区域,指引模型修正mask。
- 支持手动创建和修改多边形。
- 可对标注的多边形进行二次修改。
- 支持调整多边形之间的遮挡关系。
- 支持预览标注结果,便于用户调整。
- 支持将ISAT格式json转换为VOC、COCO、LabelMe格式,以及从COCO格式转换为ISAT格式。
安装使用步骤
假设用户已经下载了本项目的源码文件,按照以下步骤进行安装和使用:
1. 创建虚拟环境:
shell
conda create -n ISAT_with_segment_anything python==3.8
conda activate ISAT_with_segment_anything
2. 安装Segment anything:
shell
cd segment-anything
pip install -e .
cd ..
3. 安装ISAT_with_segment_anything:
shell
cd ISAT_with_segment_anything
pip install -r requirements.txt
4. 下载Segment anything预训练模型:
下载任一模型,并将模型存放于ISAT_with_segment_anything/segment_any
目录下:
- sam_vit_h_4b8939.pth
- sam_vit_l_0b3195.pth
- sam_vit_b_01ec64.pth
5. 运行软件:
shell
python main.py
下载地址
点击下载 【提取码: 4003】【解压密码: www.makuang.net】