
inpainting_gmcnn
用GAN多列卷积网络,让图像缺失区域自然重生
项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 434
使用教程
ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。
环境要求
- NVIDIA GPU
- PyTorch
- 数据集
安装与启动步骤
-
1拉仓库
论文官方实现,图像修复模型。
git clone https://github.com/shepnerd/inpainting_gmcnn.git -
2先跑测试脚本
用预训练权重验证链路,比自己训快。
sh ./script/test.sh -
3准备掩码数据
训练需要成对的掩码,格式按仓库说明。
-
4再训练
参数里指定数据集与卡号。
python train.py --dataset <数据集名> --data_file <训练清单> --gpu_ids 0
如何确认成功
给一张带遮挡的图,修复区域与周围纹理连续,没有明显的色块边界
注意事项
- 它是图像修复的经典实现,思路是可借鉴的基线。
- 预训练权重与当前框架版本可能不兼容,先试测试脚本。
- 掩码生成方式对结果影响很大,训练与推理要一致。
- 大遮挡区域仍会糊,别期待它凭空补出细节。
核心亮点
- 多列架构同时捕捉全局结构和局部纹理,修复效果更协调
- 提供PyTorch和TensorFlow双版本,方便不同框架用户使用
- 基于NeurIPS论文,算法经过学术验证,具有理论保障
不足之处
- 项目较早,依赖旧版库,环境配置可能较繁琐
- 文档偏学术,缺少快速上手的示例和API说明
适用场景
- 老照片划痕或破损区域的修复
- 图像中物体或人物的智能移除与背景填充
- 学术研究中图像修复算法的对比基准
替代项目
PartialConv、EdgeConnect、DeepFillv2
下一篇:ConSinGAN
同类项目推荐
stable-diffusion-server
开源
一键部署你的 Stable Diffusion 图像生成服务
Deploy Your Own Stable Diffusion Service
RAT-GAN
开源
用文字描述,生成对应图像
A conditional GAN for text-to-image
MIGC
开源
用边界框和文字精准控制每个物体,轻松生成多实例图像。
[CVPR 2024 Highlight] MIGC and [TPAMI 2024] MIGC++ (Official Implementation)
image-multiple-angles-3d-camera
开源
拖拽相机,3D 空间任意角度生成图像
Generate images from any camera viewpoint via 3D interactive control. Drag the camer···
Blobby
开源
一键生成随机 SVG 斑点角色,为设计加点趣味
Random SVG blob characters
DirectorsConsole
开源
在无限画布上,用电影级提示词并行驱动多台 ComfyUI 生成视频。
A web application for prompt generation and multiple ComfyUI remote and local connec···
ComfyUI_Cutoff
开源
在 ComfyUI 中切断提示词干扰,让每个元素各归其位
cutoff implementation for ComfyUI
karbonized
开源
用代码写图像,自动生成和编辑,告别手动操作。
The Programable Image Editor
color-peel
开源
把颜色从形状中剥出来,让AI按你的要求精准上色。
we propose to generate a series of geometric shapes with target colors to disentangl···