| Literature DB >> 35161998 |
Shilei Lyu1,2,3,4, Yawen Zhao1,3, Ruiyao Li1,3, Zhen Li1,2,3,4, Renjie Fan1,3, Qiafeng Li1.
Abstract
Florescence information monitoring is essential for strengthening orchard management activities, such as flower thinning, fruit protection, and pest control. A lightweight object recognition model using cascade fusion YOLOv4-CF is proposed, which recognizes multi-type objects in their natural environments, such as citrus buds, citrus flowers, and gray mold. The proposed model has an excellent representation capability with an improved cascade fusion network and a multi-scale feature fusion block. Moreover, separable deep convolution blocks were employed to enhance object feature information and reduce model computation. Further, channel shuffling was used to address missing recognition in the dense distribution of object groups. Finally, an embedded sensing system for recognizing citrus flowers was designed by quantitatively applying the proposed YOLOv4-CF model to an FPGA platform. The mAP@.5 of citrus buds, citrus flowers, and gray mold obtained on the server using the proposed YOLOv4-CF model was 95.03%, and the model size of YOLOv4-CF + FPGA was 5.96 MB, which was 74.57% less than the YOLOv4-CF model. The FPGA side had a frame rate of 30 FPS; thus, the embedded sensing system could meet the demands of florescence information in real-time monitoring.Entities:
Keywords: FPGA; YOLOv4; cascade fusion; embedded sensing system; florescence information monitoring
Mesh:
Year: 2022 PMID: 35161998 PMCID: PMC8839401 DOI: 10.3390/s22031255
Source DB: PubMed Journal: Sensors (Basel) ISSN: 1424-8220 Impact factor: 3.576
Figure 1Sample labeling. (a) Citrus flower (rainy), (b) Citrus flower (sunny), (c) Gray mold (rainy), (d) Gray mold (sunny).
Figure 2Mosaic data augmentation.
Datasets of citrus flowers and gray mold.
| Tag Name | Training Set | Test Set | Total |
|---|---|---|---|
| bud | 1005 | 249 | 1254 |
| citrus flower | 1223 | 315 | 1538 |
| gray mold | 615 | 141 | 765 |
Figure 3YOLOV4-CF neural network framework.
Figure 4Multi-scale block framework.
Figure 5YOLOv4-CF + FPGA object recognition embedded system process.
Figure 6YOLOv4-CF + FPGA object recognition model deployment process.
Configuration of different test equipment.
| No. | Platform | System | Configuration | Operating Environment |
|---|---|---|---|---|
| 1 | Server | Windows 10 | Intel Core i7-9700 @ 3.00 GHz Eight-core CPU, 16 GB RAM, Nvidia GeForce RTX 2060(6 GB) GPU | The test framework is TensorFlow 2.2.0, Keras 2.3.1, using CUDA 10.1 parallel computing framework with CUDNN 7.6.5 deep neural network acceleration library |
| 2 | PC | Windows 10 | Intel Core i7-8500U @ 1.80 GHz Four-core CPU, 16 GB RAM, Nvidia GeForce MX 450(4 GB) GPU | |
| 3 | FPGA | Linux | Xilinx Zynq UltraScale + MPSoC EV (ZCU104) SoC | Compile environment uses OpenCV and Xilinx AI runtimes |
Figure 7Training loss value of the two models.
Figure 8Improved structure comparison.
Test results of different models.
| Traning | Model | mAP@.5% | AP/% | Model Size/MB | |||
|---|---|---|---|---|---|---|---|
| Bud | Flower | Gray Mold | |||||
| Initial | YOLOv4-Tiny | 91.33 | 90.10 | 94.75 | 89.16 | 82.67 | 22.79 |
| YOLOv4-CF(I) | 92.94 | 91.33 | 94.56 | 92.94 | 84.67 | 23.31 | |
| YOLOv4-CF(II) | 93.91 | 91.76 | 95.32 | 94.65 | 86.33 | 23.44 | |
| YOLOv4-CF | 94.42 | 92.17 | 95.67 | 95.42 | 86.33 | 23.44 | |
| Transfer training | YOLOv4-Tiny | 93.61 | 92.17 | 95.28 | 93.39 | 86.00 | 22.79 |
| YOLOv4-CF(I) | 93.61 | 92.20 | 95.22 | 93.42 | 86.67 | 23.31 | |
| YOLOv4-CF(II) | 94.36 | 92.05 | 95.57 | 95.46 | 88.00 | 23.44 | |
| YOLOv4-CF | 95.03 | 92.89 | 96.22 | 95.97 | 89.00 | 23.44 | |
Performance of different methods.
| Method | mAP (%) | Test Time (ms) | Model Size (M) |
|---|---|---|---|
| MobileNet-v3 [ | 85.76 | 23.93 | 18.43 |
| GhostNet [ | 88.63 | 27.42 | 11.50 |
| DensNet121 [ | 92.74 | 42.12 | 39.11 |
| CSPDarkNet53_Tiny [ | 94.42 | 18.83 | 22.79 |
| CFNet (ours) | 95.03 | 17.93 | 23.44 |
Figure 9Test results of the YOLOv4-CF model and the YOLOv4-Tiny model on citrus flowers in the natural environment. (a) Sparse object (YOLOv4-Tiny), (b) Sparse object (YOLOv4-CF), (c) Intensive object (YOLOv4-Tiny), (d) Intensive object (YOLOv4-CF).
Recognition results of different platform test sets.
| Platform | Precision/% | Average | Model Size | Recognition Speed/ms | |||
|---|---|---|---|---|---|---|---|
| Bud | Flower | Gray Mold | |||||
| Server | 91.47 | 94.65 | 96.20 | 94.11 | 23.44 | 69.22 (CPU) | 33.28 (GPU) |
| FPGA | 89.22 | 93.54 | 95.73 | 92.83 | 5.96 | 58.74 | |
Figure 10YOLOv4-CF + FPGA model recognition results of citrus flowers.
Figure 11FPGA video stream test scheme.
Real-time recognition efficiency of different platforms.
| Platform | Configuration | Speed/FPS | Power/W |
|---|---|---|---|
| Server | CPU/core i7-9700(3.0 GHz Eight-core) | 15 | 75 |
| GPU/RTX 2060(6 GB) | 29 | 96 | |
| PC | CPU/core i7-8550U(1.8 GHz Quad-core) | 10 | 38 |
| GPU/MX150(4 GB) | 16 | 56 | |
| FPGA | ZCU104 | 17 | 20 |