Underwater color restoration with Sea-thru (Akkaynak & Treibitz, CVPR 2019): it inverts the physical underwater image-formation model per RGB channel using a per-pixel depth map. This repo extends the original image algorithm into an end-to-end video pipeline, estimating depth with Apple Depth Pro and adding temporal smoothing so restored clips don't flicker.
| Original | Restored |
|---|---|
 |
 |
The project is managed with uv (Python 3.12).
uv sync --extra depth On first run the ~1.9 GB Depth Pro checkpoint auto-downloads to checkpoints/.
./scripts/image_pipeline.sh ./scripts/video_pipeline.sh