Machine oil leakage detection based on an improved YOLOv4-tiny neural network Online publication date: Fri, 16-Feb-2024
by Yuehua Li; Sailin Wu; Jiahao Sun; Bin Hu
International Journal of Sensor Networks (IJSNET), Vol. 44, No. 2, 2024
Abstract: Oil leakage is a typical recurrent problem in industrial machine faults. An oil leakage detection network based on the YOLOv4-tiny is proposed. Firstly, we introduce depthwise separable convolution in the feature extraction network to decrease the parameters number. Secondly, MobileNetV3 is used to increase the depth and strengthen feature extraction ability. Finally, a gradient harmonising mechanism in classification loss is adopted to optimise the foreground and background class imbalance. The sufficient experiments show that the improved algorithm has a better result in the data set, which contains 677 oil leakage images. The detection accuracy reaches 89.67%, which is 1.36% higher than the original algorithm. The recall rate reaches 94.04%, 2.3% higher. The model size is reduced to 13.9M, 38.2% lower, and the frame rate is 45.6 fps. The improved model meets real-time requirements, and is lighter and easier to deploy on the edge device.
Existing subscribers:
Go to Inderscience Online Journals to access the Full Text of this article.
If you are not a subscriber and you just want to read the full contents of this article, buy online access here.Complimentary Subscribers, Editors or Members of the Editorial Board of the International Journal of Sensor Networks (IJSNET):
Login with your Inderscience username and password:
Want to subscribe?
A subscription gives you complete access to all articles in the current issue, as well as to all articles in the previous three years (where applicable). See our Orders page to subscribe.
If you still need assistance, please email subs@inderscience.com