README.md

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