<p>Unmanned aerial vehicle (UAV) equipped with visible light sensors offer a low-cost and efficient solution for monitoring wheat growth and disease. However, limitations from sensor resolution and flight altitude often result in low-quality imagery. Existing super-resolution (SR) methods for UAV-acquired wheat images often suffer from low feature utilization, insufficient frequency-domain representation, and high computational cost. To address this, we propose a lightweight recursive spatial frequency-domain transformer (LRSFT) with wavelet loss, which can enhance high-frequency detail recovery while maintaining low computational overhead. First, we design a recursive spatial module (RSM) to iteratively aggregate local spatial features with an expanding effective receptive field and model contour structural relationships. This generates a spatial recursive map while significantly reducing computational overhead. Second, a frequency-domain enhancement module (FEM) is employed to capture high-frequency features via a global receptive field. These spatial and frequency-domain maps are fused through residual connections to form a spatial-frequency feature map, enhancing reconstruction performance. Finally, we introduce a wavelet loss to address the insufficient learning of high-frequency information when using <InlineEquation ID="IEq1"> <EquationSource Format="TEX">\(L_1\)</EquationSource> <EquationSource Format="MATHML"><math> <msub> <mi>L</mi> <mn>1</mn> </msub> </math></EquationSource> </InlineEquation> loss, aiding the model in recovering high-frequency details such as wheat grains, awns, and leaf textures. Experimental results show that LRSFT achieves the best performance with a PSNR of up to <InlineEquation ID="IEq2"> <EquationSource Format="TEX">\(33.24\)</EquationSource> <EquationSource Format="MATHML"><math> <mrow> <mn>33.24</mn> </mrow> </math></EquationSource> </InlineEquation>dB, while reducing network parameters by <InlineEquation ID="IEq3"> <EquationSource Format="TEX">\(87.32\%\)</EquationSource> <EquationSource Format="MATHML"><math> <mrow> <mn>87.32</mn> <mo>%</mo> </mrow> </math></EquationSource> </InlineEquation> and <InlineEquation ID="IEq4"> <EquationSource Format="TEX">\(31.58\%\)</EquationSource> <EquationSource Format="MATHML"><math> <mrow> <mn>31.58</mn> <mo>%</mo> </mrow> </math></EquationSource> </InlineEquation>, computation by <InlineEquation ID="IEq5"> <EquationSource Format="TEX">\(89.08\%\)</EquationSource> <EquationSource Format="MATHML"><math> <mrow> <mn>89.08</mn> <mo>%</mo> </mrow> </math></EquationSource> </InlineEquation> and <InlineEquation ID="IEq6"> <EquationSource Format="TEX">\(22.81\%\)</EquationSource> <EquationSource Format="MATHML"><math> <mrow> <mn>22.81</mn> <mo>%</mo> </mrow> </math></EquationSource> </InlineEquation>, and inference time by <InlineEquation ID="IEq7"> <EquationSource Format="TEX">\(38.52\%\)</EquationSource> <EquationSource Format="MATHML"><math> <mrow> <mn>38.52</mn> <mo>%</mo> </mrow> </math></EquationSource> </InlineEquation> and <InlineEquation ID="IEq8"> <EquationSource Format="TEX">\(39.33\%\)</EquationSource> <EquationSource Format="MATHML"><math> <mrow> <mn>39.33</mn> <mo>%</mo> </mrow> </math></EquationSource> </InlineEquation> compared to the latest CNN-based RMSRGAN and Transformer-based PFT, respectively.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

LRSFT: UAV wheat image super-resolution via a lightweight recursive spatial frequency-domain transformer with wavelet loss

  • Zijian Gao,
  • Wenxia Bao,
  • Nian Wang,
  • Linsheng Huang,
  • Dong Liang

摘要

Unmanned aerial vehicle (UAV) equipped with visible light sensors offer a low-cost and efficient solution for monitoring wheat growth and disease. However, limitations from sensor resolution and flight altitude often result in low-quality imagery. Existing super-resolution (SR) methods for UAV-acquired wheat images often suffer from low feature utilization, insufficient frequency-domain representation, and high computational cost. To address this, we propose a lightweight recursive spatial frequency-domain transformer (LRSFT) with wavelet loss, which can enhance high-frequency detail recovery while maintaining low computational overhead. First, we design a recursive spatial module (RSM) to iteratively aggregate local spatial features with an expanding effective receptive field and model contour structural relationships. This generates a spatial recursive map while significantly reducing computational overhead. Second, a frequency-domain enhancement module (FEM) is employed to capture high-frequency features via a global receptive field. These spatial and frequency-domain maps are fused through residual connections to form a spatial-frequency feature map, enhancing reconstruction performance. Finally, we introduce a wavelet loss to address the insufficient learning of high-frequency information when using \(L_1\) L 1 loss, aiding the model in recovering high-frequency details such as wheat grains, awns, and leaf textures. Experimental results show that LRSFT achieves the best performance with a PSNR of up to \(33.24\) 33.24 dB, while reducing network parameters by \(87.32\%\) 87.32 % and \(31.58\%\) 31.58 % , computation by \(89.08\%\) 89.08 % and \(22.81\%\) 22.81 % , and inference time by \(38.52\%\) 38.52 % and \(39.33\%\) 39.33 % compared to the latest CNN-based RMSRGAN and Transformer-based PFT, respectively.