xinntao / SFTGAN

CVPR18 - Recovering Realistic Texture in Image Super-resolution by Deep Spatial Feature Transform

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

hi about SFT layer

shz1314 opened this issue · comments

I am sorry to trouble you, i want to introduce your SFT layer in my image-to-image translation work.If you have already finished SFT layer in pytorch, please send it to me. Very very thanks. My email is haoze_song_jnu@163.com
thanks
haoze

@shz1314, I have just added raw sft-gan testing codes to pytorch repo. Please see the sft-arch.py file in BasicSR repo.

Note that pre-training is import to sft-gan. In our implementation, we first train a normal SR network (with BN layers) under GAN architecture and then we substitute the BN layers with SFT layers (use the BN parameters as the initialization) [BN layers can also be kept and insert new SFT layers].

We will add the training codes in pytorch later. Also, we will explore more stable and effective training.