200万张雾霾图像数据集,按类型分类后针对性去雾,效果更优。
HazeSpace2M: A Dataset for Haze Aware Single Image Dehazing
- 先分雾霾类型再用专用模型去雾,提升精度
- 相比通用模型,PSNR提升2.41%,SSIM提升17.14%
- 适合需要精准去雾的自动驾驶与遥感应用
大气雾霾降低图像清晰度,对计算机视觉至关重要。现有合成数据集缺乏真实多样雾霾类型,限制了去雾算法发展。本文提出HazeSpace2M数据集,包含超过200万张图像,涵盖10种雾霾强度,覆盖雾、云和环境雾霾(EH)三类。基于该数据集,提出先分类雾霾类型再使用特定去雾模型的方法。实验显示,ResNet50与AlexNet在合成数据上准确率达92.75%和92.50%,但在真实雾霾测试集(RHT)上分别降至80%和70%,凸显数据集挑战性。采用该框架后,相比通用去雾模型,PSNR提升2.41%,SSIM提升17.14%,MSE下降10.2%。此外,该框架显著提升现有SOTA去雾模型性能。完整代码与数据集已开源于GitHub。
原文摘要 · Abstract (English)
Reducing the atmospheric haze and enhancing image clarity is crucial for computer vision applications. The lack of real-life hazy ground truth images necessitates synthetic datasets, which often lack diverse haze types, impeding effective haze type classification and dehazing algorithm selection. This research introduces the HazeSpace2M dataset, a collection of over 2 million images designed to enhance dehazing through haze type classification. HazeSpace2M includes diverse scenes with 10 haze intensity levels, featuring Fog, Cloud, and Environmental Haze (EH). Using the dataset, we introduce a technique of haze type classification followed by specialized dehazers to clear hazy images. Unlike conventional methods, our approach classifies haze types before applying type-specific dehazing, improving clarity in real-life hazy images. Benchmarking with state-of-the-art (SOTA) models, ResNet50 and AlexNet achieve 92.75\% and 92.50\% accuracy, respectively, against existing synthetic datasets. However, these models achieve only 80% and 70% accuracy, respectively, against our Real Hazy Testset (RHT), highlighting the challenging nature of our HazeSpace2M dataset. Additional experiments show that haze type classification followed by specialized dehazing improves results by 2.41% in PSNR, 17.14% in SSIM, and 10.2\% in MSE over general dehazers. Moreover, when testing with SOTA dehazing models, we found that applying our proposed framework significantly improves their performance. These results underscore the significance of HazeSpace2M and our proposed framework in addressing atmospheric haze in multimedia processing. Complete code and dataset is available on \href{https://github.com/tanvirnwu/HazeSpace2M} {\textcolor{blue}{\textbf{GitHub}}}.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。