---
license: apache-2.0
tags:
- mars
- mars-bench
- planetary-science
- computer-vision
- image-classification
---

# surface-classification

## Model Description

Vision Transformer model for Mars surface classification

**Task:** Image Classification

**Original Checkpoint:** `SurfaceClassification_ViT.ckpt`

## Mars-Bench

This model is part of the Mars-Bench benchmark, a comprehensive suite of models for Mars surface analysis tasks.

## Usage

```python
# Load the model checkpoint
import torch

model = torch.load('model.ckpt')
```

## Citation

If you use this model, please cite the Mars-Bench paper:

```bibtex
@article{marsbench2024,
  title={Mars-Bench: A Benchmark for Mars Surface Analysis},
  author={Your Citation Here},
  year={2024}
}
```

## License

This model is released under the Apache 2.0 license.