Multiple object detection of workpieces based on fusion of deep learning and image processing

Lei, Yi and Yao, Xifan and Chen, Wocheng and Zhang, Junming and Mehnen, Jorn and Yang, Erfu (2020) Multiple object detection of workpieces based on fusion of deep learning and image processing. In: IEEE World Congress on Computational Intelligence 2020, 2020-07-19 - 2020-07-24. (In Press)

[thumbnail of Lei-etal-WCCI-2020-Multiple-object-detection-of-workpieces-based-on-fusion]
Preview
Text. Filename: Lei_etal_WCCI_2020_Multiple_object_detection_of_workpieces_based_on_fusion.pdf
Accepted Author Manuscript

Download (568kB)| Preview

Abstract

A workpiece detection method based on fusion of deep learning and image processing is proposed. Firstly, the workpiece bounding boxes are located in the workpiece images by YOLOv3, whose parameters are compressed by an improved convolutional neural network residual structure pruning strategy. Then, the workpiece images are cropped based on the bounding boxes with cropping biases. Finally, the contours and suitable gripping points of the workpieces are obtained through image processing. The experimental results show that mean Average Precision (mAP) is 98.60% for YOLOv3, and 99.38% for that one by pruning 50.89% of its parameters, and the inference time is shortened by 31.13%. Image processing effectively corrects the bounding boxes obtained by deep learning, and obtains workpiece contour and gripping point information.