File size: 4,751 Bytes
d827732
 
bc3a255
 
 
d827732
 
bc3a255
 
d827732
 
18e1325
 
 
 
 
 
 
 
 
 
 
7851bd7
18e1325
 
 
 
 
 
 
 
7851bd7
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
---
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`](https://huggingface.co/docs/timm/en/index) model name  | Sentinel-1 only                                                                     | Sentinel-2 only                                                                     | Sentinel-1 and Sentinel-2                                                            |
|:-----------------|:---------------------------------------------------------------------------|:------------------------------------------------------------------------------------|:------------------------------------------------------------------------------------|:-------------------------------------------------------------------------------------|
| ConvMixer-768/32 | `convmixer_768_32`                                                         | [link](https://huggingface.co/BIFOLD-BigEarthNetv2-0/convmixer_768_32-s1-v0.1.1)    | [link](https://huggingface.co/BIFOLD-BigEarthNetv2-0/convmixer_768_32-s2-v0.1.1)    | [link](https://huggingface.co/BIFOLD-BigEarthNetv2-0/convmixer_768_32-all-v0.1.1)    |
| ConvNext v2 Base | `convnextv2_base`                                                          | [link](https://huggingface.co/BIFOLD-BigEarthNetv2-0/convnextv2_base-s1-v0.1.1)     | [link](https://huggingface.co/BIFOLD-BigEarthNetv2-0/convnextv2_base-s2-v0.1.1)     | [link](https://huggingface.co/BIFOLD-BigEarthNetv2-0/convnextv2_base-all-v0.1.1)     |
| MLP-Mixer Base   | `mixer_b16_224`                                                            | [link](https://huggingface.co/BIFOLD-BigEarthNetv2-0/mixer_b16_224-s1-v0.1.1)       | [link](https://huggingface.co/BIFOLD-BigEarthNetv2-0/mixer_b16_224-s2-v0.1.1)       | [link](https://huggingface.co/BIFOLD-BigEarthNetv2-0/mixer_b16_224-all-v0.1.1)       |
| MobileViT-S      | `mobilevit_s`                                                              | [link](https://huggingface.co/BIFOLD-BigEarthNetv2-0/mobilevit_s-s1-v0.1.1)         | [link](https://huggingface.co/BIFOLD-BigEarthNetv2-0/mobilevit_s-s2-v0.1.1)         | [link](https://huggingface.co/BIFOLD-BigEarthNetv2-0/mobilevit_s-all-v0.1.1)         |
| ResNet-50        | `resnet50`                                                                 | [link](https://huggingface.co/BIFOLD-BigEarthNetv2-0/resnet50-s1-v0.1.1)            | [link](https://huggingface.co/BIFOLD-BigEarthNetv2-0/resnet50-s2-v0.1.1)            | [link](https://huggingface.co/BIFOLD-BigEarthNetv2-0/resnet50-all-v0.1.1)            |
| ResNet-101       | `resnet101`                                                                | [link](https://huggingface.co/BIFOLD-BigEarthNetv2-0/resnet101-s1-v0.1.1)           | [link](https://huggingface.co/BIFOLD-BigEarthNetv2-0/resnet101-s2-v0.1.1)           | [link](https://huggingface.co/BIFOLD-BigEarthNetv2-0/resnet101-all-v0.1.1)           |
| ViT Base         | `vit_base_patch8_224`                                                      | [link](https://huggingface.co/BIFOLD-BigEarthNetv2-0/vit_base_patch8_224-s1-v0.1.1) | [link](https://huggingface.co/BIFOLD-BigEarthNetv2-0/vit_base_patch8_224-s2-v0.1.1) | [link](https://huggingface.co/BIFOLD-BigEarthNetv2-0/vit_base_patch8_224-all-v0.1.1) |