Project Icon

inpaint-anything

AI驱动智能图像编辑 点击实现精准修复

Inpaint Anything是一款结合Segment Anything分割技术和稳定扩散修复能力的AI图像编辑工具。通过简单点击创建精确蒙版,提高修复效率和质量。支持SAM 2、SAM-HQ和FastSAM等多种模型,适用于多种图像类型。工具提供蒙版扩展、裁剪和迭代修复等调整选项,实现灵活强大的图像编辑。

任意区域修复(使用 Segment Anything 进行修复)

任意区域修复在浏览器界面上使用 Segment Anything 输出的任意选定蒙版执行稳定扩散修复。

使用 Segment Anything 使用户能够通过简单指向所需区域来指定蒙版,而无需手动填充。这可以提高蒙版创建过程的效率和准确性,从而可能产生更高质量的修复结果,同时节省时间和精力。

AUTOMATIC1111 的 Web UI 扩展版本

说明图片

安装

请按照以下步骤安装软件:

  • 创建新的 conda 环境:
conda create -n inpaint python=3.10
conda activate inpaint
  • 克隆软件仓库:
git clone https://github.com/Uminosachi/inpaint-anything.git
cd inpaint-anything
  • 对于 CUDA 环境,安装以下软件包:
pip install -r requirements.txt
  • 如果您使用的是 macOS,请改为从以下文件安装软件包:
pip install -r requirements_mac.txt

运行应用程序

python iasam_app.py
  • 在浏览器中打开 http://127.0.0.1:7860/。
  • 注意:如果您在网络浏览器中启用了隐私保护扩展程序(如 DuckDuckGo),您可能无法从草图中检索蒙版。

选项

  • --save-seg:保存由 SAM 生成的分割图像。
  • --offline:使用离线网络执行修复。
  • --sam-cpu:在 CPU 上执行 Segment Anything 操作。

下载模型

使用方法

  • 将图像拖放到输入图像区域。
    • 通过"填充选项"可以实现外部修复,配置比例和平衡,然后点击"运行填充"按钮。
    • "动漫风格"复选框可增强分割蒙版检测,特别是在动漫风格图像中,但会稍微降低蒙版质量。
  • 点击"运行 Segment Anything"按钮。
  • 使用素描指向您想要修复的区域。您可以撤销并调整笔尺寸。
    • 将鼠标悬停在 SAM 图像或蒙版图像上,按 S 键进入全屏模式,或按 R 键重置缩放。
  • 点击"创建蒙版"按钮。蒙版将出现在所选蒙版图像区域。

蒙版调整

  • "扩展蒙版区域"按钮:用于略微扩大蒙版区域以获得更广泛的覆盖。
  • "通过素描修剪蒙版"按钮:点击此按钮将从蒙版中排除素描区域。
  • "通过素描添加蒙版"按钮:点击此按钮将素描区域添加到蒙版中。

修复选项卡

  • 输入您想要的提示和负面提示,然后选择修复模型 ID。
  • 点击"运行修复"按钮(请注意,首次下载模型可能需要一些时间)。
    • 在高级选项中,您可以调整采样器、采样步骤、指导比例和种子。
    • 如果启用"仅蒙版区域"选项,修改将仅限于指定的蒙版区域。
  • 调整迭代滑块以使用不同的种子多次执行修复。
  • 修复过程由 diffusers 提供支持。

提示

  • 您可以直接将修复后的图像拖放到 Web UI 上的输入图像字段中。(适用于 Chrome 和 Edge 浏览器)

模型缓存

  • 保存在HuggingFace缓存中且其repo_id包含"inpainting"(不区分大小写)的修复模型也会被添加到修复模型ID下拉列表中。
    • 如果您想使用特定模型,可以提前使用以下Python命令进行缓存:
    python
    
    from diffusers import StableDiffusionInpaintPipeline
    pipe = StableDiffusionInpaintPipeline.from_pretrained("Uminosachi/dreamshaper_5-inpainting")
    exit()
    
  • diffusers下载的模型通常存储在您的主目录中。Linux和MacOS用户可以在/home/用户名/.cache/huggingface/hub找到,Windows用户可以在C:\Users\用户名\.cache\huggingface\hub找到。
    • 执行修复时,如果控制台输出以下错误,请尝试从上述缓存文件夹中删除相应的模型:
    获取模型名称时发生错误...
    

清理器选项卡

  • 选择清理器模型ID。
  • 点击"运行清理器"按钮(请注意,首次下载模型可能需要一些时间)。
  • 清理过程使用Lama Cleaner执行。

仅蒙版选项卡

  • 提供仅保存蒙版而不进行其他处理的功能,以便在其他图形应用程序中使用蒙版。
  • "获取图像alpha通道蒙版"按钮:将蒙版保存为RGBA图像,蒙版保存在输入图像的alpha通道中。
  • "获取蒙版"按钮:将蒙版保存为RGB图像。

UI图像

自动保存图像

  • 修复后的图像将自动保存在outputs目录中与当前日期匹配的文件夹中。

开发

使用Inpaint Anything库,您可以执行分割并使用其他应用程序的草图创建蒙版。

许可证

源代码根据Apache 2.0许可证授权。

参考文献

  • Ravi, N., Gabeur, V., Hu, Y.-T., Hu, R., Ryali, C., Ma, T., Khedr, H., Rädel, R., Rolland, C., Gustafson, L., Mintun, E., Pan, J., Alwala, K. V., Carion, N., Wu, C.-Y., Girshick, R., Dollár, P., & Feichtenhofer, C. (2024). SAM 2: 分割图像和视频中的任何内容。arXiv预印本。
  • Kirillov, A., Mintun, E., Ravi, N., Mao, H., Rolland, C., Gustafson, L., Xiao, T., Whitehead, S., Berg, A. C., Lo, W-Y., Dollár, P., & Girshick, R. (2023). 分割任何内容。arXiv:2304.02643。
  • Ke, L., Ye, M., Danelljan, M., Liu, Y., Tai, Y-W., Tang, C-K., & Yu, F. (2023). 高质量分割任何内容。arXiv:2306.01567。
  • Zhao, X., Ding, W., An, Y., Du, Y., Yu, T., Li, M., Tang, M., & Wang, J. (2023). 快速分割任何内容。arXiv:2306.12156 [cs.CV]。
  • Zhang, C., Han, D., Qiao, Y., Kim, J. U., Bae, S-H., Lee, S., & Hong, C. S. (2023). 更快的分割任何内容:面向移动应用的轻量级SAM。arXiv:2306.14289。
项目侧边栏1项目侧边栏2
推荐项目
Project Cover

豆包MarsCode

豆包 MarsCode 是一款革命性的编程助手,通过AI技术提供代码补全、单测生成、代码解释和智能问答等功能,支持100+编程语言,与主流编辑器无缝集成,显著提升开发效率和代码质量。

Project Cover

AI写歌

Suno AI是一个革命性的AI音乐创作平台,能在短短30秒内帮助用户创作出一首完整的歌曲。无论是寻找创作灵感还是需要快速制作音乐,Suno AI都是音乐爱好者和专业人士的理想选择。

Project Cover

有言AI

有言平台提供一站式AIGC视频创作解决方案,通过智能技术简化视频制作流程。无论是企业宣传还是个人分享,有言都能帮助用户快速、轻松地制作出专业级别的视频内容。

Project Cover

Kimi

Kimi AI助手提供多语言对话支持,能够阅读和理解用户上传的文件内容,解析网页信息,并结合搜索结果为用户提供详尽的答案。无论是日常咨询还是专业问题,Kimi都能以友好、专业的方式提供帮助。

Project Cover

阿里绘蛙

绘蛙是阿里巴巴集团推出的革命性AI电商营销平台。利用尖端人工智能技术,为商家提供一键生成商品图和营销文案的服务,显著提升内容创作效率和营销效果。适用于淘宝、天猫等电商平台,让商品第一时间被种草。

Project Cover

吐司

探索Tensor.Art平台的独特AI模型,免费访问各种图像生成与AI训练工具,从Stable Diffusion等基础模型开始,轻松实现创新图像生成。体验前沿的AI技术,推动个人和企业的创新发展。

Project Cover

SubCat字幕猫

SubCat字幕猫APP是一款创新的视频播放器,它将改变您观看视频的方式!SubCat结合了先进的人工智能技术,为您提供即时视频字幕翻译,无论是本地视频还是网络流媒体,让您轻松享受各种语言的内容。

Project Cover

美间AI

美间AI创意设计平台,利用前沿AI技术,为设计师和营销人员提供一站式设计解决方案。从智能海报到3D效果图,再到文案生成,美间让创意设计更简单、更高效。

Project Cover

AIWritePaper论文写作

AIWritePaper论文写作是一站式AI论文写作辅助工具,简化了选题、文献检索至论文撰写的整个过程。通过简单设定,平台可快速生成高质量论文大纲和全文,配合图表、参考文献等一应俱全,同时提供开题报告和答辩PPT等增值服务,保障数据安全,有效提升写作效率和论文质量。

投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号