OceanSAR-1-tengeop / README.md
Balocre's picture
Update README.md
caedcc7 verified
|
raw
history blame contribute delete
4.25 kB
---
library_name: transformers
tags:
- tengeop
- SAR
- EO
- regression
- sentinel-1
- ocean
- wave-height
- earth-observation
- remote-sensing
- satellite-imagery
- synthetic-aperture-radar
- foundation-model
- linear-probing
- oceanography
- marine-forecasting
- open-source
- ocean-wind
license: apache-2.0
pipeline_tag: image-classification
base_model:
- galeio-research/OceanSAR-1
---
# Model Card for OceanSAR-1-TenGeoP
## Model Details
<img src="OceanSAR-1-logo.png" width=400>
### Model Description
OceanSAR-1-TenGeoP is a linear probing head for classifying ocean geophysical phenomena, built on top of the OceanSAR-1 foundation model. It leverages the powerful features extracted by OceanSAR-1 to accurately identify 10 different geophysical phenomena in Synthetic Aperture Radar (SAR) imagery.
- **Developed by:** Thomas Kerdreux, Alexandre Tuel @ [Galeio](http://galeio.fr)
- **Deployed by:** Antoine Audras @ [Galeio](http://galeio.fr)
- **Model type:** Linear Classification Head on Vision Foundation Model
- **License:** Apache License 2.0
- **Base model:** OceanSAR-1 (ResNet50/ViT variants)
- **Training data:** Sentinel-1 Wave Mode (WV) SAR images with labeled geophysical phenomena
## Uses
### Direct Use
This model is designed for automated classification of geophysical phenomena in SAR imagery over ocean surfaces. It can be used for:
- Rapid identification of ocean features in SAR data
- Monitoring of maritime environments
- Automated analysis of large SAR datasets
- Ocean science and research applications
### Performance Results
The model achieves state-of-the-art performance on TenGeoP classification, with performance varying by backbone architecture:
| Backbone | TenGeoP Accuracy (%) |
|----------|---------------------|
| ResNet50 | 75.5 |
| ViT-S/16 | 78.6 |
| ViT-S/8 | 82.1 |
| ViT-B/8 | 83.6 |
## How to Use
```python
import torch
from transformers import AutoModelForImageClassification
# Load the foundation model and classification head
oceansar = AutoModelForImageClassification.from_pretrained("galeio-research/OceanSAR-1-tengeop")
# Prepare your SAR image (should be single-channel VV polarization)
dummy_image = torch.randn(1, 1, 256, 256) # (B, C, H, W)
# Extract features and classify geophysical phenomena
with torch.no_grad():
outputs = oceansar(dummy_image)
predicted_class = torch.argmax(outputs.logits, dim=1).item()
```
## Training Details
### Training Data
- **Dataset:** Sentinel-1 Wave Mode (WV) SAR images with labeled geophysical phenomena
- **Labels:** 10 classes of ocean geophysical phenomena
- **Size:** Balanced dataset across all classes
- **Preprocessing:** Same as base OceanSAR-1 model
## Evaluation
### Metrics
TenGeoP classification performance is evaluated using accuracy (%), achieving:
- 75.5% accuracy with ResNet50 backbone
- 78.6% accuracy with ViT-S/16 backbone
- 82.1% accuracy with ViT-S/8 backbone
- 83.6% accuracy with ViT-B/8 backbone
### Comparison to Other Backbones
The model outperforms existing approaches:
- CROMA (ViT-B/8): 65.4% accuracy
- MoCo (ResNet50): 60.9% accuracy
- DeCUR (ResNet50): 58.3% accuracy
- DOFA (ViT-B/16): 58.4% accuracy
- DOFA (ViT-L/16): 63.4% accuracy
- SoftCon (ViT-S/14): 73.2% accuracy
- SoftCon (ViT-B/14): 74.8% accuracy
## Technical Specifications
### Hardware Requirements
- Same as base model
- Minimal additional computational cost for inference
### Dependencies
- PyTorch >= 1.8.0
- Transformers >= 4.30.0
- Base OceanSAR-1 model
### Input Specifications
- Same as base OceanSAR-1 model
- Single channel (VV polarization) SAR images
- 256x256 pixel resolution
## Citation
**BibTeX:**
```bibtex
@article{kerdreux2025efficientselfsupervisedlearningearth,
title={Efficient Self-Supervised Learning for Earth Observation via Dynamic Dataset Curation},
author={Kerdreux, Thomas and Tuel, Alexandre and Febvre, Quentin and Mouche, Alexis and Chapron, Bertrand},
journal={arXiv preprint arXiv:2504.06962},
year={2025},
eprint={2504.06962},
archivePrefix={arXiv},
primaryClass={cs.CV},
url={https://arxiv.org/abs/2504.06962},
}
```
## Acknowledgements
This work was granted access to the HPC resources of IDRIS and TGCC under the allocation 2025-[A0171015666] made by GENCI.