WeatherDiffusion 项目推荐
1. 项目基础介绍和主要编程语言
WeatherDiffusion 是一个开源项目,专注于使用基于补丁的去噪扩散模型来恢复恶劣天气条件下的视觉图像。该项目的主要编程语言是 Python。它提供了一个代码库,用于训练和执行推理,以在恶劣天气条件下进行图像恢复。
2. 项目的核心功能
WeatherDiffusion 的核心功能包括:
- 图像去雪:通过去噪扩散模型去除图像中的雪。
- 图像去雨和去雾:结合去雨和去雾技术,恢复在雨天和雾天拍摄的图像。
- 雨滴去除:专门用于去除图像中的雨滴。
- 多天气条件下的图像恢复:支持在多种恶劣天气条件下的图像恢复,使用 AllWeather 训练集进行多天气恢复。
3. 项目最近更新的功能
最近更新的功能包括:
- 预训练模型分享:分享了两个预训练模型 WeatherDiff64 和 WeatherDiff128,分别用于不同分辨率的图像恢复。
- 评估脚本优化:提供了优化的评估脚本
eval_diffusion.py
,支持不同的测试集和采样步数。 - 可视化工具:增加了可视化工具,展示了补丁基础的扩散图像恢复过程,包括去雪、去雨和去雾的效果。
通过这些更新,WeatherDiffusion 进一步提升了在恶劣天气条件下图像恢复的效果和应用范围。