Mirali33's picture
Upload README.md with huggingface_hub
276d061 verified
metadata
license: apache-2.0
tags:
  - mars
  - mars-bench
  - planetary-science
  - computer-vision
  - image-segmentation

crater-binary-segmentation

Model Description

U-Net model for binary crater segmentation

Task: Image Segmentation

Original Checkpoint: CraterBinarySegmentation_UNet.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.