文件名称：partialconv:重新实现“使用部分卷积对不规则Kong进行图像修补”
文件大小：3.12MB
文件格式：ZIP
更新时间：2024-05-24 13:00:35
Python
PyTorch部分卷积修复
这是“使用部分卷积对不规则Kong进行图像修补”的重新实现。
这不是作者的正式实现。
结果
从上到下，输入图像，蒙版图像，原始输出，具有除蒙版区域以外的地面真相的输出和地面真相图像。
设置
克隆此仓库
git clone https://github.com/tanimutomo/partialconv.git
安装要求
所需的库以编写。
快速运行
下载预训练的模型
您可以下载预，该模型与输出上述图像的模型相同。 请将预训练的模型文件partialconv/ 。
跑步
快速运行
python predict.py
您可以指定图像/蒙版/模型路径
python predict.py --img
【文件预览】：
partialconv-master
----.gitignore(1KB)
----experiment.py(4KB)
----requirements.txt(86B)
----src()
--------__init__.py(0B)
--------utils.py(2KB)
--------generate_mask.py(5KB)
--------loss.py(5KB)
--------dataset.py(2KB)
--------evaluate.py(731B)
--------model.py(8KB)
--------train.py(3KB)
----figs()
--------pconv_result.png(3.02MB)
----examples()
--------mask4.png(3KB)
--------mask1.png(5KB)
--------mask2.png(3KB)
--------img2.jpg(17KB)
--------img0.jpg(15KB)
--------img1.jpg(8KB)
--------img3.jpg(12KB)
--------img4.jpg(21KB)
--------mask3.png(5KB)
--------mask0.png(6KB)
----README.md(2KB)
----predict.py(2KB)
----default_config.yml(1KB)