PixelwiseRegressionTask#

class torchgeo.trainers.PixelwiseRegressionTask(model='resnet50', backbone='resnet50', weights=None, in_channels=3, num_outputs=1, labels=None, num_filters=3, loss='mse', lr=0.001, patience=10, freeze_backbone=False, freeze_decoder=False)[source]#

Bases: RegressionTask

LightningModule for pixelwise regression of images.

Added in version 0.5.

configure_models()[source]#

Initialize the model.