.png){kind=icon}
Return
Improved Lightweight Mango Sorting Model Based on Visualization
.png){kind=icon}
OA
AI
DOI：10.3390/agriculture12091467.png)
.png){kind=icon}
.png){kind=icon}
Share.png){kind=icon}
Save
Abstract
En 中文
Neural networks are widely used in fruit sorting and have achieved some success. However, due to the limitations of storage space and power consumption, the storage and computing of a neural network model on embedded devices remain a massive challenge. Aiming at realizing a lightweight mango sorting model, the feature-extraction characteristics of the shallow and deep networks of the SqueezeNet model were analyzed by a visualization method, and then eight lightweight models were constructed by removing redundant layers or modifying the convolution kernel. It was found that the model designated Model 4 performed well after training and testing. The class activation mapping method was used to explain the basis of the classification decision, and the model was compared with ten classical classification models. The results showed that the calculation performance of the model was significantly improved without reducing accuracy. The parameter storage requirement is 0.87 MB, and the calculation amount is 181 MFLOPS, while the average classification accuracy can still be maintained at 95.64%. This model has a high-cost performance and can be widely used in embedded devices.
Keywords:
deep learning
lightweight convolutional neural network
visualization
mango sorting
Journal
IF:
3.6
Papers:
1.3W
Citations:
2.9W
Organization
Cited Papers
ShareSave
Cobalt and copper complexes with formamidine ligands: Synthesis, crystal X-ray study, DFT calculations and cytotoxicity.png)
Polyhedron
PolyhedronIF0
PREAI
ShareSaveShareSaveDetection and classification of chilli leaf disease using a squeeze-and-excitation-based CNN model
PREAI
ShareSaveShareSave
View more