athena.layers.resnet_block

an implementation of resnet block

Module Contents

Classes

ResnetBasicBlock

Basic block of resnet

class athena.layers.resnet_block.ResnetBasicBlock(num_filter, stride=1)

Bases: tensorflow.keras.layers.Layer

Basic block of resnet Reference to paper “Deep residual learning for image recognition”

call(self, inputs)

call model

make_downsample_layer(self, num_filter, stride)

perform downsampling using conv layer with stride != 1