基于加权随机森林的图像超分辨率算法研究

吴成东¹，卢紫微^2,3，于晓升¹

1. 东北大学机器人科学与工程学院2. 东北大学信息科学与工程学院3. 辽宁石油化工大学计算机与通信工程学院

摘 要：针对目前图像超分辨率重建效果欠佳的问题,提出一种基于加权随机森林的图像超分辨率重建算法.利用随机森林对图像块的特征进行聚类,并引入岭回归模型建立每类叶子结点中高、低分辨率图像块的映射关系,重建时根据测试低分辨率图像块所属的类别以及在每类叶子结点中的K近邻近似拟合误差,进行加权预测获得高分辨率图像块.将图像的非局部自相似性与迭代反投影算法相结合对预测的高分辨率图像进行后处理以提高重建质量.实验结果表明,所提出算法可以有效提高峰值信噪比,具有较好的可视效果.

关键词：超分辨率;随机森林;岭回归;K近邻;加权预测;