metadata
license: apache-2.0
tags:
- mars
- mars-bench
- planetary-science
- computer-vision
- image-classification
frost-classification
Model Description
Swin Transformer model for frost classification on Mars
Task: Image Classification
Original Checkpoint: FrostClassification_SwinTransformer.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.