README / README.md
hackelle's picture
Update README.md
7851bd7 verified
|
raw
history blame
4.75 kB
metadata
title: README
emoji: πŸš€
colorFrom: blue
colorTo: green
sdk: static
pinned: false
license: mit
short_description: Official Repository of Pretrained Models on BigEarthNet v2.0

BigEarthNetv2.0 Pretrained Weights

We provide pretrained weights for several different models. All models were trained with different seeds. The weights for the best-performing model (based on Macro Average Precision on the recommended test split) are uploaded. All models are available as versions using Sentinel-1 only, Sentinel-2 only or Sentinel-1 and Sentinel-2 data.

The order of bands is as follows:
For Models using Sentinel-1 only: ["VH", "VV"]
For Models using Sentinel-2 only: 10m bands, 20m bands = ["B02", "B03", "B04", "B08", "B05", "B06", "B07", "B11", "B12", "B8A"]
For Models using Sentinel-1 and Sentinel-2: 10m bands, 20m bands, S1 bands = ["B02", "B03", "B04", "B08", "B05", "B06", "B07", "B11", "B12", "B8A", "VH", "VV"]

The output classes are always in alphabetical order:
['Agro-forestry areas', 'Arable land', 'Beaches, dunes, sands', 'Broad-leaved forest', 'Coastal wetlands', 'Complex cultivation patterns', 'Coniferous forest', 'Industrial or commercial units', 'Inland waters', 'Inland wetlands', 'Land principally occupied by agriculture, with significant areas of natural vegetation', 'Marine waters', 'Mixed forest', 'Moors, heathland and sclerophyllous vegetation', 'Natural grassland and sparsely vegetated areas', 'Pastures', 'Permanent crops', 'Transitional woodland, shrub', 'Urban fabric']

Model equivalent timm model name Sentinel-1 only Sentinel-2 only Sentinel-1 and Sentinel-2
ConvMixer-768/32 convmixer_768_32 link link link
ConvNext v2 Base convnextv2_base link link link
MLP-Mixer Base mixer_b16_224 link link link
MobileViT-S mobilevit_s link link link
ResNet-50 resnet50 link link link
ResNet-101 resnet101 link link link
ViT Base vit_base_patch8_224 link link link