Skip to main content

一个基于 SAM2 和 Tkinter 的交互式图像分割与标注工具。

Project description

SAM2 交互式图像标注工具

PyPI version Python versions License

一个基于 Meta 的 Segment Anything Model 2 (SAM2) 和 Tkinter 构建的图形化、交互式图像分割与标注工具。它旨在提供一个简单易用、开箱即用的方式,利用强大的AI模型来辅助进行高精度的图像标注工作。

(注意:上面的徽章在您将包成功发布到 PyPI 后才会生效。)

主要功能

  • ✨ 智能分割: 集成强大的 SAM2 模型,通过点击正/负点进行高质量的掩码预测,极大提升标注效率。
  • ✏️ 手动微调: 支持手动绘制多边形进行精确标注,或对AI生成的掩码进行补充和修正。
  • 📝 标签编辑: 方便地对已完成的标注区域进行标签修改、删除和管理。
  • 🖱️ 多模式操作: 在 "SAM标注"、"多边形" 和 "编辑标签" 模式间自由切换,应对不同标注场景。
  • 🖼️ 便捷的视图控制: 支持图像的无级缩放与平移,轻松处理高分辨率大图。
  • 💾 会话管理: 自动保存和加载上一次的标注会话,方便随时中断和继续工作。
  • 📄 格式兼容: 以与 LabelMe 兼容的 JSON 格式保存标注结果,便于下游任务处理。
  • 🚀 自动下载: 首次运行时自动从网络下载所需的模型权重,无需手动配置。

安装

安装过程分为两个必要步骤。请务必按顺序执行。

⚠️ 步骤 1: 手动下载并安装前提依赖 segment-anything-2

本工具依赖于 Meta 官方的 segment-anything-2 库。由于网络原因,直接通过 pip 从 GitHub 安装可能会失败,我们推荐采用更稳定的手动下载安装方式。

  1. 访问官方仓库: 请前往 segment-anything-2 的官方 GitHub 仓库: https://github.com/facebookresearch/segment-anything-2

  2. 下载代码(推荐使用 git clone: 打开您的终端,cd 到一个您方便存放代码的目录,然后运行以下命令来克隆完整的代码仓库。这是最可靠的方式。

    git clone [https://github.com/facebookresearch/segment-anything-2.git](https://github.com/facebookresearch/segment-anything-2.git)
    

    (备选方案:您也可以在 GitHub 页面上点击 Code -> Download ZIP 来下载,但 git clone 是更专业的做法。)

  3. 从本地安装: 下载完成后,进入刚刚创建的文件夹,并使用 pip 进行本地安装。

    # 进入下载好的文件夹
    cd segment-anything-2
    
    # 从当前目录安装
    pip install .
    

步骤 2: 安装本工具 EEBD-GESAM2

完成上一步后,你就可以通过 PyPI 直接安装本工具了:

pip install EEBD-GESAM2

注意:在首次启动本工具时,程序会自动从网络下载所需的 SAM2 模型权重文件(大小可能为几GB),此过程可能需要一些时间,具体取决于你的网络状况。下载完成后,模型将被缓存,未来启动时将不再需要下载。

快速开始

安装完成后,你的系统中就有了一个新的命令行指令。直接在终端中运行它即可启动程序:

eebd-gesam2

程序启动后,会显示图形界面。你可以通过点击 "加载图像" 按钮来选择一个包含多张图片的文件夹,然后开始你的标注工作。

许可证

本项目采用 MIT 许可证。详情请见 LICENSE 文件。

致谢

本项目的开发离不开以下这些优秀的项目和社区:


Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

eebd_gesam2-0.1.3.tar.gz (21.0 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

eebd_gesam2-0.1.3-py3-none-any.whl (22.2 kB view details)

Uploaded Python 3

File details

Details for the file eebd_gesam2-0.1.3.tar.gz.

File metadata

  • Download URL: eebd_gesam2-0.1.3.tar.gz
  • Upload date:
  • Size: 21.0 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.10.16

File hashes

Hashes for eebd_gesam2-0.1.3.tar.gz
Algorithm Hash digest
SHA256 feff2e9fa6bb148399c47606da238e0c0f612104d8f775c2122f13b890c12b02
MD5 05beba6be9afffa70c4aa3eda44395e1
BLAKE2b-256 76d2b5088c1ab99d26876351671e5d0f06f255af2bec08ee8b144c6848eeaf1f

See more details on using hashes here.

File details

Details for the file eebd_gesam2-0.1.3-py3-none-any.whl.

File metadata

  • Download URL: eebd_gesam2-0.1.3-py3-none-any.whl
  • Upload date:
  • Size: 22.2 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.1.0 CPython/3.10.16

File hashes

Hashes for eebd_gesam2-0.1.3-py3-none-any.whl
Algorithm Hash digest
SHA256 813a9d6c7a0fb2bbcbac03a57f844d590595e124cc1b182d15fe78c03a138f35
MD5 90d330ad0c1c4221f9ad88324719a4d5
BLAKE2b-256 7235747a2bbe6c6cead6713c07c89ab2fa6277c01cbb24ed2e947e43ff222364

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page