FCN#

class torchgeo.models.FCN(in_channels, classes, num_filters=64)[source]#

Bases: Module

A simple 5 layer FCN with leaky relus and ‘same’ padding.

__init__(in_channels, classes, num_filters=64)[source]#

Initializes the 5 layer FCN model.

Parameters:
  • in_channels (int) – Number of input channels that the model will expect

  • classes (int) – Number of filters in the final layer

  • num_filters (int) – Number of filters in each convolutional layer

forward(x)[source]#

Forward pass of the model.