Offical code for paper:"SiENet: Siamese Expansion Network for Image Extrapolation". IEEE Signal Processing Letters
In this paper, we propose a novel two-stage siamese adversarial model for image extrapolation, named Siamese Expansion Network(SiENet). In two stages, a novel border sensitive convolution named filling convolution is designed for allowing encoder to predict the unknown content, alleviating the burden of decoder. Besides, to introduce prior knowledge to network and reinforce the inferring ability of encoder, siamese adversarial mechanism is designed to enable our network to model covered long range feature toward the uncovered image feature.
1.Pytorch >= 1.0
2.Python 3
3.NVIDIA GPU + CUDA 9.0
4.Tensorboard
5.Matlab
1.Clone the code
git clone https://github.com/nanjingxiaobawang/SieNet-Image-extrapolation
2.Build Gaussian Sampling CUDA package
cd ./SieNet-Image-extrapolation/resample2d_package
python setup.py install --user
our method is evaluated on three datasets, i.e., Cityscapes, paris street-view and beach . Single-direction evaluation is made on Scenery dataset. Smooth image could be get by structureflow
You can download these datasets from their project website.
1.CelebA
3.beach
4.scenery
You need to generate mask, We give a mask for your reference.
To train our model, modify the model config file model_config.yaml. You may need to change the path of dataset or the parameters of the networks etc. Then run the following code:
python train.py --name=[the name of your experiment] --path=[path save the results]
you can achieve pretrained model here: Paris-streetview, Cityscapes you can use the test.py to generate results. Run the following code:
python test.py --name=[the name of your experiment] --path=[path of your experiments] --input=[input images] --mask=[mask images] --structure=[structure images] --output=[path to save the output images] --model=[which model to be tested]
Please kindly cite the following paper in your publications if it helps your research:
title={SiENet: Siamese Expansion Network for Image Extrapolation},
author={ Zhang, X. and Chen, F. and Wang, C. and Tao, M. and Jiang, G. },
journal={IEEE Signal Processing Letters},
volume={PP},
number={99},
pages={1-1},
year={2020},
}