cyto_dl.nn.spatial_transformer module#

class cyto_dl.nn.spatial_transformer.ConvPoolReLU(in_filters, out_filters, kernel_size)[source]#

Bases: Module

forward(x)[source]#
class cyto_dl.nn.spatial_transformer.STN(n_input_ch=2, patch_shape=(64, 256, 512), n_conv_filters=32)[source]#

Bases: Module

forward(x)[source]#
toggle(direction)[source]#