Upload README.md with huggingface_hub

README.md

@@ -0,0 +1,39 @@
+---
+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.