基于BAS-2D-VMD的图像去噪算法

Image Denoising Algorithm Based on BAS-2D-VMD

针对2D-VMD(2D-Variational Mode Decomposition)算法需要人为预设K值的问题,提出了一种基于BAS-2D-VMD(Beetle Antennae Search Algorithm-2D-Variational Mode Decomposition)的图像去噪算法。该方法首先用BAS算法搜寻2D-VMD参数K的最佳值,然后对图像进行2D-VMD分解,得到K个子模态,最后将对计算各个模态与原图像的相关系数进行模态筛选,并将选中的模态进行重构,得到去噪后的图像。仿真实验表明,该方法与中值滤波和维纳滤波相比取得了较高的峰值信噪比。将该算法应用到架空输电线图像处理领域,对无人机采集到的线路图像进行去噪,与传统的算法对比取得较好的去噪效果,解决了线路图像噪声难以去除的问题。

An image denoising algorithm based on BAS-2D-VMD(Beetle Antennae Search Algorithm-2D-Variational Mode Decomposition)is proposed for the defect that the 2D-VMD(2D-Variational Mode Decomposition)algorithm needs artificially preset K value.This method first uses the BAS algorithm to search for the optimal value of the 2D-VMD parameter K,and then decomposes the image on 2D-VMD to obtain K sub-modes.Mode screening is performed by calculating the correlation coefficient of each mode.Finally,the remaining modes to an image is reconstructed.The image quality is judged by calculating the peak signal-to-noise ratio,and the denoising effect is verified by simulation.This algorithm is applied in the field of image processing of overhead transmission lines,and denoising line images collected by the UAV(Unmanned Aerial Vehicle).Compared with the traditional median filtering and wavelet threshold denoising algorithms,a better denoising effect is obtained.The problem that the line image noise is difficult to remove is solved.