Abstract
Recently, deep neural networks have led the progress of instance segmentation. There are several models of neural networks, such as the Mask R-CNN network, that perform this task with good results. However, it is possible to improve the results of the Mask R-CNN network if several models of it are combined, and their results are merged using Ensemble Learning. We propose the algorithm “Simple Instance Segmentation Ensemble” that is capable of assembling the results of two Mask R-CNN networks to produce better results. In our experiments, we train several Mask R-CNN networks with synthetic images of machinery objects. In addition, these Mask R-CNN networks have different backbones and different sizes of kernels for the Gaussian Blur filter applied to the synthetic machinery images used during training. We tested the performance of these networks by predicting real images of machinery. Besides, we propose the SISE algorithm to assemble the predictions of two previously trained Mask R-CNN networks, and we obtained better results than those of the individual Mask R-CNN networks. In particular, our best result is an ensemble that has one Mask R-CNN trained with synthetic images smoothed by Gaussian Blur filter with a kernel size of 7 \(\times \) 7, and another network with a kernel size of 3 \(\times \) 3. Both networks have as backbone a ResNeXt 101 with FPN (Feature Pyramid Network). This ensemble has a bounding box mAP of 89.42% and a segmentation mAP of 88.34% in the real machinery test images.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Chen, X., Girshick, R., He, K., Dollár, P.: Tensormask: a foundation for dense object segmentation. In: Proceedings of the IEEE/CVF International Conference on Computer Vision, pp. 2061–2069 (2019)
Du, X., et al.: Spinenet: learning scale-permuted backbone for recognition and localization. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 11 592–11 601 (2020)
He, K., Gkioxari, G., Dollár, P., Girshick, R.: Mask R-CNN. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 2961–2969 (2017)
Hinterstoisser, S., Lepetit, V., Wohlhart, P., Konolige, K.: On pre-trained image features and synthetic images for deep learning. In: Proceedings of the European Conference on Computer Vision (ECCV) (2018)
Salas, A.J.C., Meza-Lovon, G., Fernández, M.E.L., Raposo, A.: Training with synthetic images for object detection and segmentation in real machinery images. In: 2020 33rd SIBGRAPI Conference on Graphics, Patterns and Images (SIBGRAPI). IEEE, pp. 226–233 (2020)
Mera, D., Fernández-Delgado, M., Cotos, J.M., Viqueira, J.R.R., Barro, S.: Comparison of a massive and diverse collection of ensembles and other classifiers for oil spill detection in sar satellite images. Neural Comput. Appl. 28(1), 1101–1117 (2017)
Rodriguez, J.J., Kuncheva, L.I., Alonso, C.J.: Rotation forest: A new classifier ensemble method. IEEE transactions on pattern analysis and machine intelligence 28(10), 1619–1630 (2006)
Gao, X., Shan, C., Hu, C., Niu, Z., Liu, Z.: An adaptive ensemble machine learning model for intrusion detection. IEEE Access 7, 82 512–82 521 (2019)
Gaonkar, B., et al.: Multi-parameter ensemble learning for automated vertebral body segmentation in heterogeneously acquired clinical mr images. IEEE J. Trans. Eng. Health Med. 5, 1–12 (2017)
Zuo, Y., Drummond, T.: Fast residual forests: rapid ensemble learning for semantic segmentation. In: Conference on Robot Learning. PMLR, pp. 27–36 (2017)
Kang, J., Gwak, J.: Ensemble of instance segmentation models for polyp segmentation in colonoscopy images. IEEE Access, vol. 7, pp. 26 440–26 447 (2019)
He, K., Zhang, X., Ren, S., Sun, J.: Deep residual learning for image recognition. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 770–778 (2016)
Wu, Y., Kirillov, A., Massa, F., Lo, W.-Y., Girshick, R.: Detectron2 (2019). https://github.com/facebookresearch/detectron2
Lin, T.-Y., et al.: Microsoft COCO: common objects in context. In: Fleet, D., Pajdla, T., Schiele, B., Tuytelaars, T. (eds.) ECCV 2014. LNCS, vol. 8693, pp. 740–755. Springer, Cham (2014). https://doi.org/10.1007/978-3-319-10602-1_48
Lin, T.-Y., et al.: Feature pyramid networks for object detection. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 2117–2125 (2017)
Xie, S., Girshick, R., Dollár, P., Tu, Z., He, K.: Aggregated residual transformations for deep neural networks. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 1492–1500 (2017)
Acknowledgment
M. E. LOAIZA acknowledges the financial support of the CONCYTEC – BANCO MUNDIAL Project “Mejoramiento y Ampliación de los Servicios del Sistema Nacional de Ciencia Tecnología e Innovación Tecnológica” 8682-PE, through its executing unit PROCIENCIA, within the framework of the call E041-01, Contract No. 038-2018-FONDECYT-BM-IADT-AV.
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2021 Springer Nature Switzerland AG
About this paper
Cite this paper
Cerpa, A., Meza-Lovon, G., Fernández, M.E.L. (2021). Ensemble Learning to Perform Instance Segmentation over Synthetic Data. In: Bebis, G., et al. Advances in Visual Computing. ISVC 2021. Lecture Notes in Computer Science(), vol 13018. Springer, Cham. https://doi.org/10.1007/978-3-030-90436-4_25
Download citation
DOI: https://doi.org/10.1007/978-3-030-90436-4_25
Published:
Publisher Name: Springer, Cham
Print ISBN: 978-3-030-90435-7
Online ISBN: 978-3-030-90436-4
eBook Packages: Computer ScienceComputer Science (R0)