Accurate recognition of coal-gangue image based on lightweight HPG-YOLOX-S model
Accurate recognition of coal-gangue image based on lightweight HPG-YOLOX-S model
Blog Article
The existing coal-gangue separation methods based on vision technology have problems of large model parameter amount, poor feature extraction capability and low recognition precision.In order to solve the above problems, a coal-gangue recognition method based on YOLOX-S model combined lightweight Ghost-S network and hybrid parallel attention module (HPAM) named HPG-YOLOS-S model is proposed.Firstly, HPAM is added click here to the backbone network of YOLOX-S model.Thus the important information in an image is enhanced, and the secondary information is inhibited.
The feature extraction capability of the backbone network is enhanced.Secondly, the backbone network of YOLOX-S model is replaced by Ghost-S network with smaller parameter quantity.The utilization rate and feature fusion capability are improved.Finally, in the predection layer, the SIOU loss function is used to replace the loss function of YOLOX-S model to impsrove the detection and positioning precision and enhance the extraction capability of the target.
In order to verify the detection effect of the proposed method on large coal-gangue, the HPG-YOLOX-S model is compared with YOLOX-S model.The results show that the identification accuracy of the HPG-YOLOX-S model for coal and gangue is 99.53% and 99.60% respectively, which is 2.
51% and 1.27% higher than those of YOLOX-S model.The results of validation show that the precision rate, recall rate and F1 value of the HPG-YOLOX-S model are all above 94%, which are 5.68%, 3.
51% and 2.91% higher than those of YOLOX-S model respectively.The parameters amount of the HPG-YOLOX-S model is 7.8 MB, which is 1.
2 MB lower than that of YOLOX-S model.The ablation experiment results show that the mean average precision of the HPG-YOLOX-S model is 9.17% higher than that of YOLOX-S model.The borstlist självhäftande experiment result of visualization of the thermodynamic diagram shows that the HPG-YOLOX-S model focuses on the surface differences between coal and gangue, such as texture and contour.
The model pays more attention to the overall target of coal-gangue.