README.md

metadata

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

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.