Mirali33/mars-bench-surface-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

# Load the model checkpoint
import torch

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

Citation

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

@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.