这节我们介绍利用一个预训练模型清除图像中雾霾,使图像更清晰。

26.1 导入需要的模块

26.2 查看原来的图像

26.3 定义一个神经网络

这个神经网络主要由卷积层构成,该网络将构建在预训练模型之上。

26.4 训练模型

clean_photo/test_images/shanghai02.jpg done!

26.5 查看处理后的图像

处理后的图像与原图像拼接在一起,保存在clean_photo /results目录下。

虽非十分理想,但效果还是比较明显的!
本章数据集下载地址(提取码是:1nxs)
更多内容可参考:
https://github.com/TheFairBear/PyTorch-Image-Dehazing