Segment-Anything-Easy WebUI

本项目Fork自SAM-webui

欢迎来到我为 segment-anything 所做的一个简易可本地搭建的webui demo。

🚀 安装与设置

本项目需要Node.JS点击此处下载

1. 创建本地虚拟环境并安装所需库：

本项目只支持Windows：

  - 1.对于N卡用户，请确认CUDA已安装，然后双击setup.bat文件进行安装。
  - 2.对于A卡用户，请直接双击AMD-setup.bat进行安装。（AMD显卡使用DML来加速运算，可能会有bug）

2. 下载必要的模型文件：

   • sam_vit_h_4b8939.pth

   下载好模型文件后，将其放入项目的 models 目录下。

3. 启动 等待setup.bat或AMD-setup.bat运行完成,双击run.bat或AMD-run.bat即可启动程序。

🎨 使用方法

运行程序后，在cmd的输出框里你可以找到一个* Running on http://127.0.0.1:xxxx (xxxx为端口号)*这样的地址，在浏览器打开即可

• 左侧是默认图片，一只可爱的可琳 by stable diffusion。
• 你可以通过选择文件按钮导入自己的图片。
• 通过点击左侧图片上的点来选择我们需要裁剪出来的 mask，可以多点，识别率更高。

> 🔴 红点表示你选择的点。点击 "Clear Points" 可以清除所有点。

• 选择好你想要的点后，点击 "Process Points"。稍等片刻，即可生成你的 mask。生成速度取决于你的gpu（mac使用cpu会比较慢，目前的mps模式会报错，还未解决）

• 如果这就是你想要的 mask，那么点击 "Download" 即可下载你的 mask。

📝 注意事项

• app.py 是本项目的入口。
• Python 使用 Flask，Web 前端使用 TypeScript + HTML + CSS。你所开发的所有 TypeScript 代码会在调试运行时自动转换为 JavaScript。
• 在 tools 文件夹中还有一个小工具 get_all_masks.py。它可以将 tools/input 目录下所有的图片（JPG、PNG）的所有 mask 提取出来并放入 output 文件夹中。(目前只支持N卡)
• 目前的 webui 界面还很简陋。如果你愿意，欢迎帮助完善它！

📓 TODO

• 全面支持AMD
• 增加黑白底切换
• 添加更多模型

🌟 致谢

感谢 Facebook Research 团队提供的 segment-anything 项目作为本项目的基础。

🤝 贡献

如果你有任何建议或想为这个项目做出贡献，欢迎通过 Issues 或 Pull Requests 与我联系。

📄 许可证

本项目采用 MIT License 许可。