TFSamepaddingLayer

tiatoolbox . models . architecture . hovernet . TFSamepaddingLayer

class TFSamepaddingLayer(ksize, stride)

To align with tensorflow same padding.

Putting this before any conv layer that needs padding. Here, we assume kernel has same height and width for simplicity.

Initialize TFSamepaddingLayer.

Methods

forward

Logic for using layers defined in init.

Attributes

training

Parameters:

• ksize (int)

• stride (int)

forward(x)

Logic for using layers defined in init.

Parameters:

• self (TFSamepaddingLayer)

• x (Tensor)

Return type:

Tensor