Aluminum Surface-Defect Detection Based on Multi-Task Deep Learning

Xiaohai Shen; Zehao Li; Min Li; Xiaolong Xu; Xuewu Zhang

doi:10.3788/LOP57.101501

[1] Tang B, Kong J Y, Wu S Q. Review of surface defect detection based on machine vision[J]. Journal of Image and Graphics, 22, 1640-1663(2017).

[2] Wang S, Wu X, Zhang Y H et al. Surface crack segmentation based on multi-scale wavelet transform and structured forest[J]. Acta Optica Sinica, 38, 0815024(2018).

[3] Zhang F, Wu Y, Xiao Z T et al. Nanoparticle segmentation based on U-Net convolutional neural network[J]. Laser & Optoelectronics Progress, 56, 061005(2019).

[4] Ronneberger O, Fischer P, Brox T. U-Net: convolutional networks for biomedical image segmentation[M]. ∥Navab N, Hornegger J, Wells W, et al. Medical image computing and computer-assisted intervention. Lecture notes in computer science. Cham: Springer, 9351, 234-241(2015).

[5] Zhang X W, Ding Y Q, Yan-Yun L et al. A vision inspection system for the surface defects of strongly reflected metal based on multi-class SVM[J]. Expert Systems with Applications, 38, 5930-5939(2011).

[6] Wang L S, Zhong Y, Li Z D et al. On-line fabric defect recognition algorithm based on deep learning[J]. Journal of Computer Applications, 39, 2125-2128(2019).

[7] Szegedy C, Liu W, Jia Y et al. Going deeper with convolutions. [C]∥Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, June 7-12, 2015, Boston. New York: IEEE, 15523970(2015).

[8] Chang H T, Gou J N, Li X M. Application of Faster R-CNN in image defect detection of industrial CT[J]. Journal of Image and Graphics, 23, 1061-1071(2018).

[9] Ren S, He K, Girshick R et al. Faster R-CNN: towards real-time object detection with region proposal networks. [C]∥Advances in neural information processing systems, December 7-12, 2015, Montreal, Quebec, Canada. New York: Curran Associates., 91-99(2015).

[10] Liong S T, Gan Y, Huang Y C et al. -03-28)[2019-10-18]. https:∥arxiv., org/abs/1903, 12139(2019).

[11] He K, Gkioxari G, Dollár P et al. Mask R-CNN. [C]∥Proceedings of the IEEE International Conference on Computer Vision, October 22-29, 2017, Venice, Italy. New York: IEEE, 2961-2969(2017).

[13] Zhang Y. -07-27)[2019-10-17]. https:∥arxiv.xilesou., top/abs/1707, 08114(2018).

[14] Zhang H L, Xiao L Q, Wang Y K et al. A generalized recurrent neural architecture for text classification with multi-task learning. [C]∥Proceedings of the Twenty-Sixth International Joint Conference on Artificial Intelligence, August 19-26, 2017, Melbourne, Australia. California: IJC AI, 3385-3391(2017).

[15] Wan J, Zhang Z L, Yan J W et al. Sparse Bayesian multi-task learning for predicting cognitive outcomes from neuroimaging measures in Alzheimer's disease. [C]∥2012 IEEE Conference on Computer Vision and Pattern Recognition, June 16-21, 2012, Providence, RI. New York: IEEE, 940-947(2012).

[16] Yuan C F, Hu W M, Tian G D et al. Multi-task sparse learning with beta process prior for action recognition. [C]∥2013 IEEE Conference on Computer Vision and Pattern Recognition, June 23-28, 2013, Portland, OR, USA. New York: IEEE, 423-429(2013).

[17] Zhang Y. -07-27)[2019-10-18]. https:∥arxiv., org/abs/1707, 08114(2018).

[18] He K M, Zhang X Y, Ren S Q et al. Deep residual learning for image recognition. [C]∥2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), June 27-30, 2016, Las Vegas, NV, USA. New York: IEEE, 770-778(2016).

[19] Lin T Y, Dollár P, Girshick R et al. Feature pyramid networks for object detection. [C]∥2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), July 21-26, 2017, Honolulu, HI, USA. New York: IEEE, 2117-2125(2017).

[20] Long J, Shelhamer E, Darrell T. Fully convolutional networks for semantic segmentation. [C]∥2015 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), June 7-12, 2015, Boston, MA, USA. New York: IEEE, 3431-3440(2015).

[21] Gibert X, Patel V M, Chellappa R. Deep multitask learning for railway track inspection[J]. IEEE Transactions on Intelligent Transportation Systems, 18, 153-164(2017).

[22] Yu C M, Li H N, An L. Analysis of text emotion cause based on multi-task deep learning[J]. Journal of Guangxi Normal University(Natural Science Edition), 37, 50-61(2019).

[23] Yang H X, Chen F, Gan W F. Deep face recognition algorithm based on multitask learning[J]. Laser & Optoelectronics Progress, 56, 181005(2019).

[24] Cipolla R, Gal Y, Kendall A. Multi-task learning using uncertainty to weigh losses for scene geometry and semantics. [C]∥2018 IEEE/CVF Conference on Computer Vision and Pattern Recognition, June 18-23, 2018, Salt Lake City, UT, USA. New York: IEEE, 7482-7491(2018).

[25] Inc. Faster R-CNN. 2019-10-18][EB/OL]. GitHub, Mask R-CNN in PyTorch 1.0.(2018). https://github.com/facebookresearch/maskrcnn-benchmark.

[26] -11-16)[2019-10-18]. https:∥tianchi.aliyun.com/competition/entrance/231682/information.(2018).

[27] Garcia-Garcia A, Orts-Escolano S, Oprea S et al. -04-22)[2019-10-18]. https:∥arxiv., org/abs/1704, 06857(2017).

[28] Wu X Z, Zhou Z H. A unified view of multi-label performance measures[C]∥Proceedings of the 34th International Conference on Machine Learning, August 6-11, 2017, Sydney, NSW, Australia., 3780-3788(2017).

[29] Lin T Y, Goyal P, Girshick R et al. Focal loss for dense object detection. [C]∥2017 IEEE International Conference on Computer Vision (ICCV), October 22-29, 2017, Venice. New York: IEEE, 2980-2988(2017).