diff --git a/torchgeo/models/swin.py b/torchgeo/models/swin.py index 4d720f469cd..842c62abd38 100644 --- a/torchgeo/models/swin.py +++ b/torchgeo/models/swin.py @@ -17,13 +17,15 @@ __all__ = ["Swin_V2_B_Weights"] # https://github.com/allenai/satlas/blob/bcaa968da5395f675d067613e02613a344e81415/satlas/cmd/model/train.py#L42 # noqa: E501 -# Satlas uses the TCI product for Sentinel-2 RGB, which is in the range (0, 255). +# Satlas uses the TCI product for Sentinel-2 RGB, which is in the range (0, 255). +# See details: https://github.com/allenai/satlas/blob/main/Normalization.md#sentinel-2-images. # noqa: E501 # Satlas Sentinel-1 and RGB Sentinel-2 and NAIP imagery is uint8 and is normalized to (0, 1) by dividing by 255. # noqa: E501 _satlas_transforms = AugmentationSequential( K.Normalize(mean=torch.tensor(0), std=torch.tensor(255)), data_keys=["image"] ) # Satlas uses the TCI product for Sentinel-2 RGB, which is in the range (0, 255). +# See details: https://github.com/allenai/satlas/blob/main/Normalization.md#sentinel-2-images. # noqa: E501 # Satlas Sentinel-2 multispectral imagery has first 3 bands divided by 255 and the following 6 bands by 8160, both clipped to (0, 1). # noqa: E501 _std = torch.tensor( [255.0, 255.0, 255.0, 8160.0, 8160.0, 8160.0, 8160.0, 8160.0, 8160.0]