.png){kind=icon}
返回
Improved Lightweight Mango Sorting Model Based on Visualization
.png){kind=icon}
OA
AI
DOI：10.3390/agriculture12091467.png)
.png){kind=icon}
.png){kind=icon}
分享.png){kind=icon}
收藏
摘要
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.
Keyword:
deep learning
lightweight convolutional neural network
visualization
mango sorting
期刊
IF:
3.6
论文数:
1.3W
被引数:
2.9W
机构
引用论文
Cobalt and copper complexes with formamidine ligands: Synthesis, crystal X-ray study, DFT calculations and cytotoxicity.png)
Polyhedron
PolyhedronIF0
PREAI
分享收藏分享收藏Detection and classification of chilli leaf disease using a squeeze-and-excitation-based CNN model
PREAI
分享收藏
查看更多内容