基于太赫兹成像结合压缩感知与超分辨的葵花籽饱满度检测

    Detection of Sunflower Seed Plumpness Based on Terahertz Imaging Combined with Compressive Sensing and Super-Resolution

    • 摘要: 太赫兹成像技术虽已被证实能够用于检测葵花籽内部品质,然而其成像速度较为缓慢,难以实现切实且迅速的检测。该研究将压缩感知与注意力增强超分辨率生成对抗网络(A-ESRGAN)模型相结合应用于太赫兹成像领域,实现对葵花籽饱满度的快速检测。首先,选用压缩采样匹配追踪(compressive sampling matching pursuit,CoSaMP)重构算法来验证不同测量矩阵的性能,根据最佳综合性能选取高斯(gauss)矩阵作为最终的测量矩阵。其次,通过比较基于交替方向乘子法(alternating direction method of multipliers,ADMM)结合全变分(total variation,TV)正则化(ADMM_TV)和子空间追踪(subspace pursuit,SP)等5种重构算法的峰值信噪比(PSNR)和重构时间等评价指标评估图像重建质量,结果表明ADMM_TV在PSNR、均方误差(NMSE)、结构相似性指数(SSIM)表现最佳,自然图像质量评估器(NIQE)在测量比例超过6.0%最低,尽管重构时间无明显优势,但综合表现优于其他算法。最后,运用多尺度注意力增强超分辨率生成对抗网络(A-ESRGAN-Multi)模型对压缩感知不同采样率的重构图像进行处理,其效果优于真实图像增强超分辨率生成对抗网络(Real-ESRGAN)和单尺度注意力增强超分辨率生成对抗网络(A-ESRGAN-Single),提升了图像质量,使边缘对比度得以提高,为后续的图像分割提供了便利。研究表明,压缩感知与A-ESRGAN-Multi模型相结合用于检测葵花籽饱满度是可行的,验证集的饱满度误差平均为2.502%,最大检测误差为6.41%。综上所述,将压缩感知与A-ESRGAN-Multi模型相结合,能够有效地节省82.5%的采样时间以及存储空间,为葵花籽的品质检测开辟了新的途径,提供了全新的技术支持。

       

      Abstract: As a highly promising non-contact detection means, terahertz imaging technology has gradually come to the fore in recent years and has been conclusively proven to possess the ability to detect the internal quality of sunflower seeds. Terahertz waves can penetrate the shells of sunflower seeds and interact with their internal tissue structures and components, thereby obtaining information regarding their quality. However, the application of this technology has not been all plain sailing. The problem of its slow imaging speed is like an insurmountable chasm, severely impeding its effective application in large-scale and rapid detection scenarios. Traditional terahertz imaging methods usually entail consuming a substantial amount of time to collect and process data, which makes them seem inadequate when faced with the task of detecting the quality of sunflower seeds that requires practical and prompt detection, and thus unable to meet the high-efficiency requirements in industrial production or actual agricultural detection. To overcome this difficulty, this paper proposes a method that combines compressed sensing with the Attention-Enhanced Super-Resolution Generative Adversarial Network (A-ESRGAN) model and applies it to the field of terahertz imaging, aiming to achieve rapid and accurate detection of the plumpness of sunflower seeds. In the initial stage of the research, we focused on the crucial step of selecting the measurement matrix in compressed sensing. Firstly, the Compressive Sampling Matching Pursuit (CoSaMP) reconstruction algorithm was initially adopted to verify the performance of different measurement matrices. Based on the best overall performance, the Gauss matrix was selected as the final measurement matrix. Secondly, by comparing the Peak Signal-to-Noise Ratio (PSNR), reconstruction time and other evaluation indicators of five reconstruction algorithms such as the Alternating Direction Method of Multipliers (ADMM) combined with Total Variation (TV) regularization (ADMM_TV) and Subspace Pursuit (SP), the results showed that ADMM_TV performed best in terms of PSNR, Mean Squared Error (NMSE), and Structural Similarity Index (SSIM). The Natural Image Quality Evaluator (NIQE) had the lowest value when the measurement ratio exceeded 6.0%. Although it had no obvious advantage in reconstruction time, its overall performance was better than that of other algorithms. Therefore, ADMM_TV was chosen. Finally, the Attention-Enhanced Super-Resolution Generative Adversarial Network with Multiple Scales (A-ESRGAN-Multi) model was utilized to process the reconstructed images of different sampling rates in compressed sensing. Its effect was better than that of the Real-ESRGAN and the Attention-Enhanced Super-Resolution Generative Adversarial Network with a Single Scale (A-ESRGAN-Single), enhancing the image quality and increasing the edge contrast, which provided convenience for subsequent image segmentation. The research indicates that the combination of compressed sensing and the A-ESRGAN-Multi model for detecting the fullness of sunflower seeds is feasible. The average plumpness error on the validation set is 2.502%, and the maximum detection error is 6.41%.In conclusion, the combination of compressed sensing and the A-ESRGAN-Multi model can effectively save 82.5% of the sampling time and storage space. With the reduction of data volume, not only is the pressure on storage devices alleviated, reducing the hardware cost, but also the management and transmission of data become more convenient and efficient, thus opening up a new path and providing new technical support for the quality detection of sunflower seeds.

       

    /

    返回文章
    返回