
基于AODNet的单幅图像去雾方法——利用PyTorch实现
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
本研究提出了一种基于AODNet的单幅图像去雾算法,并使用Python深度学习框架PyTorch进行实现。该方法有效提升了去雾效果,使处理后的图像更加清晰自然。
基于AODNet的图像雾度去除
使用Pytorch实现Boyi Li在ICCV 2017上的论文《AOD-Net: Real-Time Single Image Haze Removal Using Dark Channel Prior》中提出的单幅图像去雾霾技术。
### 内容和依赖性:
* Python版本:3.6或更高
* 环境配置:Torch==1.7.1,Pillow==5.1.0,numpy==1.14.3,matplotlib==2.2.2
### 使用方法:
下载整个项目后运行inference.py文件。
#### 文件结构说明:
- `.saved_models` 文件夹:用于保存训练好的模型(格式为.pth)。
- `datagt` 文件夹:存放无雾图像的ground truth数据。
- `datahazy` 文件夹:包含与训练数据对应的模糊图像。
- `test_images` 文件夹:包括原论文中使用的一些测试图片。
#### 关键文件:
* data.py: 包含加载训练所需的数据函数。
* train.py: 使用位于.data文件夹中的训练资料从头开始训练新的AODNet模型。
* model.py: AODNet的定义及相关操作。
全部评论 (0)
还没有任何评论哟~


