GitHub 热门项目：BrushNet

GitHub项目：BrushNet 仓库地址：https://github.com/TencentARC/BrushNet 星级：1745 | 作者：腾讯ARC 项目描述：[ECCV 2024] 论文《BrushNet: A Plug-and-Play Image Inpainting Model with Decomposed Dual-Branch Diffusion》正式实现 =================================================== 自述文件内容： # 刷网 该存储库包含 ECCV2024 论文“BrushNet：具有分解双分支扩散的即插即用图像修复模型”的实现 关键词：图像修复、扩散模型、图像生成 > [鞠轩](https://github.com/juxuan27)¹²、[刘贤](https://alvinliu0.github.io/)¹²、[王心涛](https://xinntao.github.io/)^1*、[雨轩]卞](https://scholar.google.com.hk/itations?user=HzemVzoAAAAJ&hl=zh-CN&oi=ao)²、[单颖](https://www.linkedin.com/in/YingShanProfile/)¹、[徐强](https://cure-lab.github.io/)^2*
> ¹腾讯PCG ARC实验室²香港中文大学^*通讯作者 🌐项目页面 | 📜Arxiv | 🗄️数据 | 📹视频 | 🤗拥抱脸部演示 |

**📖 目录** - [画笔网](#brushnet) - [🔥 更新日志](#-update-log) - [待办事项](#待办事项) - [🛠️方法概述](#️-方法概述) - [🚀 入门](#-getting-started) - [环境要求🌍](#environment-requirement-) - [数据下载⬇️](#data-download-️) - [🏃🏼 运行脚本](#-running-scripts) - [训练🤯](#training-) - [推论📜](#inference-) - [评价📏](#evaluation-) - [🤝🏼 引用我们](#-cite-us) - [💖 致谢](#-致谢) ## 🔥 更新日志 - [2024/12/17] 📢 📢 [BrushEdit](https://github.com/TencentARC/BrushEdit) 发布，一款高效、白盒、自由格式的图像编辑工具，由 LLM-agents 和一体化修复模型支持。 - [2024/12/17] 📢 📢 [BrushNetX](https://huggingface.co/TencentARC/BrushEdit/tree/main/brushnetX)（更强的BrushNet）模型发布。 ## 待办事项 - [x] 发布训练和推理代码 - [x] 发布检查点 (sdv1.5) - [x] 释放检查点 (sdxl)。遗憾的是，我只有 V100 来训练这个检查点，它只能以批量大小 1 进行训练，速度很慢。当前的 ckpt 仅针对较小的步数进行训练，因此表现不佳。但幸运的是，[yuanhang](https://github.com/yuanhangangio) 自愿帮助训练更好的版本。请继续关注！感谢[yuanhang](https://github.com/yuanhangangio)的努力！ - [x] 发布评估代码 - [x] 发布 gradio 演示 - [x] 发布 comfyui 演示。感谢 [nullquant](https://github.com/nullquant) ([ConfyUI-BrushNet](https://github.com/nullquant/ComfyUI-BrushNet)) 和 [kijai](https://github.com/kijai) ([ComfyUI-BrushNet-Wrapper](https://github.com/kijai/ComfyUI-BrushNet-Wrapper)) 的帮助！ - [x] 发布[训练数据](https://huggingface.co/datasets/random123123/BrushData)。感谢 [random123123](https://huggingface.co/random123123) 的帮助！ - [x