search

yccyenchicheng / pytorch-SegInpaint

A PyTorch Implementation of Segmentation for Image Inpainting based on SPG-Net: https://arxiv.org/abs/1805.03356
19 stars 2 forks source link

Do you have test code? #3

Open jialiang66 opened 4 years ago

jialiang66 commented 4 years ago

As the title

yccyenchicheng commented 4 years ago

Sorry I did not provide a separate file for testing. I might add that later in my spare time.

However, you could modify model.forward(...) for inference. I hope that will not cause too much trouble for you.

Thank you.