taki0112 / SRM-Tensorflow

Simple Tensorflow implementation of "SRM : A Style-based Recalibration Module for Convolutional Neural Networks"

SRM-Tensorflow

Simple Tensorflow implementation of "SRM : A Style-based Recalibration Module for Convolutional Neural Networks"

Usage

from SRM import SRM_block

x = SRM_block(x, channels, use_bias=False, is_training=is_training, scope='srm_block')

How to use in residual block

Results

Classification

Style Transfer

Reference

• SRMNet-Pytorch

Author

Junho Kim