| license: apache-2.0 | |
| tags: | |
| - segment-anything | |
| - sam | |
| - uwsam | |
| - underwater-segmentation | |
| # UWSAM Teacher Model | |
| This repository contains the trained checkpoints for the UWSAM Teacher model. | |
| ## Model Checkpoints | |
| The following checkpoints are available: | |
| - `checkpoints/epoch_4.pth` | |
| - `checkpoints/epoch_5.pth` | |
| - `checkpoints/epoch_6.pth` | |
| ## Usage | |
| ```python | |
| from huggingface_hub import hf_hub_download | |
| # Download a specific checkpoint | |
| checkpoint_path = hf_hub_download( | |
| repo_id="fhaosss/UWSAM_teacher", | |
| filename="checkpoints/epoch_6.pth" | |
| ) | |
| ``` | |
| ## Training Configuration | |
| The training configuration is available in `config/uwsam_teacher_paper.py`. | |
| ## Citation | |
| If you use this model, please cite the original UWSAM paper. | |