Li-Xiaoguang / MISF

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

MISF:Multi-level Interactive Siamese Filtering for High-Fidelity Image Inpainting

We proposed a novel approach for high-fidelity image inpainting. Specifically, we use a single predictive network to conduct predictive filtering at the image level and deep feature level, simultaneously. The image-level filtering is to recover details, while the deep feature-level filtering is to complete semantic information, which leads to high-fidelity inpainting results. Our method outperforms state-of-the-art methods on three public datasets.[ArXiv]

Framework

Dataset

  1. For data folder path (CelebA) organize them as following:
--CelebA
   --train
      --1-1.png
   --valid
      --1-1.png
   --test
      --1-1.png
   --mask-train
	  --1-1.png
   --mask-valid
      --1-1.png
   --mask-test
      --0%-20%
        --1-1.png
      --20%-40%
        --1-1.png
      --40%-60%
        --1-1.png
  1. Run the code ./data/data_list.py to generate the data list

Pretrained models

CelebA

Places2

Dunhuang

Train

python train.py
For the parameters: checkpoints/config.yml, kpn/config.py

Test

python test.py
For the parameters: checkpoints/config.yml, kpn/config.py

Results

  • Comparsion with SOTA, see paper for details.

Framework

More details are coming soon

Bibtex

Acknowledgments

Parts of this code were derived from:
https://github.com/tsingqguo/efficientderain
https://github.com/knazeri/edge-connect

About


Languages

Language:Python 100.0%