/ Модель

Характеристики

Лицензия
MIT
Библиотека
configilm
Опубликована
11 окт 2024

mobilevit_s-s2-v0.2.0

Классификация изображенийИзображение → label3 профессии

MobileViT-S, обученная на BigEarthNet v2.0, классифицирует мультиспектральные спутниковые снимки Sentinel-2.

48скачиваний за 30 дней0лайковвеса обновлены22 мая 2025

Бенчмарки

Как модель справилась с задачами пользователей ailist и что о ней публикуют открытые лидерборды

Публичные оценки

Из открытых лидербордов. Сравнивайте результаты одного теста при одинаковых условиях — отсутствие оценки не значит, что модель хуже.

В подключённых источниках результатов для этой модели пока нет.

Что умеет модель

Описание составлено по карточке модели и метаданным Hugging Face

Это модель MobileViT-S, обученная на наборе BigEarthNet v2.0 (reBEN) для классификации мультиспектральных снимков Sentinel-2. Модель предназначена для анализа спутниковых изображений и отнесения наблюдений к классам земного покрова, представленным в наборе данных.

Веса опубликованы после 20 эпох обучения. В README приведены результаты на тестовой выборке BigEarthNet v2.0: macro average precision 0,704 и macro F1 0,621; микроусреднённые показатели выше. Модель использует библиотеку ConfigILM и распространяется в формате safetensors по лицензии MIT.

Сценарии применения

  • классификация спутниковых снимков Sentinel-2
  • определение типов земного покрова по мультиспектральным данным
  • анализ и сравнение результатов классификации на BigEarthNet v2.0

Ограничения

Оценка качества приведена на тестовой выборке BigEarthNet v2.0; перенос качества на другие спутниковые данные и схемы классов не подтверждён. Метрики macro ниже micro, поэтому качество может различаться между классами.

Для каких профессий подходит3

Разделы каталога профессий, в задачах которых используется эта модель

Все профессии

Карточка модели

README от автора на Hugging Face, как он его опубликовал

Открыть на Hugging Face
TU BerlinRSiMDIMABigEarthBIFOLD

Mobilevit_s pretrained on BigEarthNet v2.0 using Sentinel-2 bands

This model was trained on the BigEarthNet v2.0 (also known as reBEN) dataset using the Sentinel-2 bands. It was trained using the following parameters:

  • Number of epochs: up to 100 (with early stopping after 5 epochs of no improvement based on validation average precision macro)
  • Batch size: 512
  • Learning rate: 0.001
  • Dropout rate: 0.15
  • Drop Path rate: 0.15
  • Learning rate scheduler: LinearWarmupCosineAnnealing for 1000 warmup steps
  • Optimizer: AdamW
  • Seed: 0

The weights published in this model card were obtained after 20 training epochs. For more information, please visit the official BigEarthNet v2.0 (reBEN) repository, where you can find the training scripts.

BigEarthNet

The model was evaluated on the test set of the BigEarthNet v2.0 dataset with the following results:

MetricMacroMicro
Average Precision0.7042230.861498
F1 Score0.6208510.757460
Precision0.7457290.811422

Example

A Sentinel-2 image (true color representation)
BigEarthNet
Class labelsPredicted scores
Agro-forestry areas Arable land Beaches, dunes, sands ... Urban fabric 0.000000 0.000000 0.000000 ... 1.000000

To use the model, download the codes that define the model architecture from the official BigEarthNet v2.0 (reBEN) repository and load the model using the code below. Note that you have to install configilm to use the provided code.

from reben_publication.BigEarthNetv2_0_ImageClassifier import BigEarthNetv2_0_ImageClassifier

model = BigEarthNetv2_0_ImageClassifier.from_pretrained("path_to/huggingface_model_folder")

e.g.

from reben_publication.BigEarthNetv2_0_ImageClassifier import BigEarthNetv2_0_ImageClassifier

model = BigEarthNetv2_0_ImageClassifier.from_pretrained(
  "BIFOLD-BigEarthNetv2-0/mobilevit_s-s2-v0.2.0")

If you use this model in your research or the provided code, please cite the following papers:

K. Clasen, L. Hackel, T. Burgert, G. Sumbul, B. Demir, V. Markl, "reBEN: Refined BigEarthNet Dataset for Remote Sensing Image Analysis", IEEE International Geoscience and Remote Sensing Symposium (IGARSS), 2025.

@inproceedings{clasen2025refinedbigearthnet,
  title={{reBEN}: Refined BigEarthNet Dataset for Remote Sensing Image Analysis},
  author={Clasen, Kai Norman and Hackel, Leonard and Burgert, Tom and Sumbul, Gencer and Demir, Beg{"u}m and Markl, Volker},
  year={2025},
  booktitle={IEEE International Geoscience and Remote Sensing Symposium (IGARSS)},
}

L. Hackel, K. Clasen, B. Demir, "ConfigILM: A General Purpose Configurable Library for Combining Image and Language Models for Visual Question Answering.", SoftwareX 26 (2024): 101731.

@article{hackel2024configilm,
  title={ConfigILM: A general purpose configurable library for combining image and language models for visual question answering},
  author={Hackel, Leonard and Clasen, Kai Norman and Demir, Beg{"u}m},
  journal={SoftwareX},
  volume={26},
  pages={101731},
  year={2024},
  publisher={Elsevier}
}

Как запустить

На своей GPU в ailist. Цена — по текущим предложениям провайдеров

GPU в ailist

Универсальный рантайм · configilm · image-classification

Нельзя запустить

Архитектура ? неизвестна transformers в образе рантайма.

Ревизия весов
766d511

0 моделей