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