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

    Detection of sunflower seed plumpness based on Terahertz imaging combined with compressive sensing and super-resolution

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

       

      Abstract: Terahertz imaging has been one of the most promising technologies for non-contact detection in recent years. Among them, the internal quality of sunflower seeds can be detected using terahertz imaging. Specifically, terahertz waves can penetrate the shells of sunflower seeds, and then interact with their internal tissue structures and components, thereby obtaining their information of quality. However, the imaging speed has severely impeded the effective application of terahertz imaging in large-scale and rapid detection scenarios. Traditional terahertz imaging can usually consume a substantial amount of time to collect and then process the image data. Thus, it is highly required to accurately and rapidly detect the quality of sunflower seeds, in order to fully meet the high-efficiency requirements in industrial production or actual detection in modern agriculture. In this study, the compressed sensing with the Attention-Enhanced Super-Resolution Generative Adversarial Network (A-ESRGAN) model was proposed in the field of terahertz imaging. Rapid and accurate detection was also achieved in the plumpness of sunflower seeds. Furthermore, the crucial step was to select the measurement matrix in the initial stage during compressed sensing. Firstly, the Compressive Sampling Matching Pursuit (CoSaMP) reconstruction was adopted to verify the performance of different measurement matrices. According to the best overall performance, the Gauss matrix was then selected as the final measurement matrix. Secondly, the Peak Signal-to-Noise Ratio (PSNR) and reconstruction time were also compared using five reconstruction algorithms, such as the Alternating Direction Method of Multipliers (ADMM) combined with Total Variation (TV) regularization (ADMM_TV) and Subspace Pursuit (SP). Evaluation indicators show that the ADMM_TV performed best, in terms of the PSNR, Mean Squared Error (NMSE), and Structural Similarity Index (SSIM). The Natural Image Quality Evaluator (NIQE) shared the lowest value when the measurement ratio exceeded 6.0%. Although there was no outstanding advantage in the reconstruction time, the overall performance of the ADMM_TV algorithm was better than that of the rest. Finally, the Attention-Enhanced Super-Resolution Generative Adversarial Network with Multiple Scales (A-ESRGAN-Multi) model was utilized to process the reconstructed images with the different sampling rates during compressed sensing. A better performance was then achieved, compared with the Real-ESRGAN and the Attention-Enhanced Super-Resolution Generative Adversarial Network with a Single Scale (A-ESRGAN-Single). The image quality and the edge contrast were enhanced for the subsequent image segmentation. Consequently, it was very feasible to combine the compressed sensing and the A-ESRGAN-Multi model, in order to detect the fullness of sunflower seeds. The average error of plumpness on the validation set was only 2.50%, and the maximum error of detection was 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. Data volume was also reduced to alleviate the pressure on the storage devices and hardware costs during data transmission and processing. Thus, the finding can also provide new technical support to the quality detection of sunflower seeds.

       

    /

    返回文章
    返回