Unofficial PyTorch implementation starter for SnowMaster: Comprehensive Real-world Image Desnowing via MLLM with Multi-Model Feedback Optimization (CVPR 2025).
If this repo saves you reading / reproduction time, please star it and follow @StaryMoon. I am building honest open reproduction starters for recent CVPR papers.
This repository is an independent, unofficial, work-in-progress starter.
- Paper: SnowMaster: Comprehensive Real-world Image Desnowing via MLLM with Multi-Model Feedback Optimization
- Venue: CVPR 2025
- Reproduction status: benchmarks are not reproduced yet
- Relationship to authors: this repo is not official and is not affiliated with the paper authors.
This v0.1.0 starter implements a compact, readable scaffold inspired by the paper:
- compact image encoder and decoder
- task/prompt conditioning tokens
- residual restoration head
- toy L1 + prompt-consistency loss
- smoke-test script
The goal is to make the high-level idea easy to inspect, fork, and improve.
- exact paper architecture
- official checkpoints
- dataset-specific training pipeline
- PSNR / SSIM benchmark reproduction
git clone https://github.com/StaryMoon/SnowMaster-Unofficial.git
cd SnowMaster-Unofficial
pip install -r requirements.txt
python scripts/smoke_test.pyExpected output includes:
loss: ...
restored: torch.Size([2, 3, 64, 64])
import torch
from snowmaster_unofficial import UnofficialStarter
image = torch.rand(2, 3, 64, 64)
model = UnofficialStarter(kind="restoration")
out = model(image)- Replace toy modules with a closer implementation of the paper.
- Add dataset loader and config files.
- Add metric scripts and visualization.
- Reproduce a small benchmark or ablation table.
- Add pretrained weights once experiments are stable.
cvpr-2025, image-desnowing, mllm, image-restoration, pytorch, unofficial-implementation
Please cite the original paper if you use the method. This repo is only an unofficial starter and does not replace the paper.
MIT License. The original paper and official materials remain owned by their respective authors / publishers.