项目作者: kostyaev

项目描述 :
Convolution NN resize initialization for subpixel convolutions
高级语言: Jupyter Notebook
项目地址: git://github.com/kostyaev/ICNR.git
创建时间: 2017-11-16T10:05:37Z
项目社区:https://github.com/kostyaev/ICNR

开源协议:MIT License

下载


Checkerboard artifact free sub pixel convolution initialization

Tensorflow implementation of ICNR initialization used in https://arxiv.org/pdf/1707.02937.pdf

screenshot

Updated for TF v2.4.

Usage example:

Wrap up your initialization with ICRN and you are ready to go:

  1. layers.Conv2DTranspose(..., strides=scale, kernel_initializer=ICNR(GlorotUniform(), scale))