littlebot
Published on 2025-04-14 / 1 Visits
0

【源码】基于PyQt5框架的交互式图像标注工具

项目简介

本项目名为"ISAT with segment anything",是基于PyQt5框架开发的交互式图像标注工具。它集成了"segment anything"模型,可实现图像分割的快速标注,支持语义分割和实例分割。具备交互式修正mask、手动创建多边形、二次修改、调整多边形遮挡关系、预览标注结果等功能。此外,还支持将ISAT格式json转换为VOC、COCO、LabelMe格式,以及从COCO格式转换为ISAT格式。

项目的主要特性和功能

  1. 集成"segment anything"模型,实现交互式半自动图像分割标注。
  2. 支持通过鼠标左(右)键点击感兴趣(不感兴趣)区域,指引模型修正mask。
  3. 支持手动创建和修改多边形。
  4. 可对标注的多边形进行二次修改。
  5. 支持调整多边形之间的遮挡关系。
  6. 支持预览标注结果,便于用户调整。
  7. 支持将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 anythingshell cd segment-anything pip install -e . cd .. 3. 安装ISAT_with_segment_anythingshell 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】