Skip to main content

Dense Material Segmentation with Context-Aware Network

  • Conference paper
  • First Online:
Computer Vision, Imaging and Computer Graphics Theory and Applications (VISIGRAPP 2022)

Abstract

The dense material segmentation task aims at recognising the material for every pixel in daily images. It is beneficial to applications such as robot manipulation and spatial audio synthesis. Modern deep-learning methods combine material features with contextual features. Material features can generalise to unseen images regardless of appearance properties such as material shape and colour. Contextual features can reduce the segmentation uncertainty by providing extra global or semi-global information about the image. Recent studies proposed to crop the images into patches, which forces the network to learn material features from local visual clues. Typical contextual information includes extracted feature maps from networks targeting object and place related tasks. However, due to the lack of contextual labels, existing methods use pre-trained networks to provide contextual features. As a consequence, the trained networks do not give a promising performance. Their accuracy is below 70%, and the predicted segments have coarse boundaries. Considering this problem, this chapter introduces the Context-Aware Material Segmentation Network (CAM-SegNet). The CAM-SegNet is a hybrid network architecture to simultaneously learn from contextual and material features jointly with labelled materials. The effectiveness of the CAM-SegNet is demonstrated by training the network to learn boundary-related contextual features. Since the existing material datasets are sparsely labelled, a self-training approach is adopted to fill in the unlabelled pixels. Experiments show that CAM-SegNet can identify materials correctly, even with similar appearances. The network improves the pixel accuracy by 3–20% and raises the Mean IoU by 6–28%.

Supported by the University of Southampton.

This is a preview of subscription content, log in via an institution to check access.

Access this chapter

Chapter
USD 29.95
Price excludes VAT (USA)
  • Available as PDF
  • Read on any device
  • Instant download
  • Own it forever
eBook
USD 64.99
Price excludes VAT (USA)
  • Available as EPUB and PDF
  • Read on any device
  • Instant download
  • Own it forever
Softcover Book
USD 84.99
Price excludes VAT (USA)
  • Compact, lightweight edition
  • Dispatched in 3 to 5 business days
  • Free shipping worldwide - see info

Tax calculation will be finalised at checkout

Purchases are for personal use only

Institutional subscriptions

Notes

  1. 1.

    https://vision.cs.uiuc.edu/attributes/.

  2. 2.

    https://kylezheng.org/research-projects/densesegattobj/.

  3. 3.

    http://opensurfaces.cs.cornell.edu/.

  4. 4.

    http://opensurfaces.cs.cornell.edu/publications/minc/.

  5. 5.

    https://vision.ist.i.kyoto-u.ac.jp/codeanddata/localmatdb/.

  6. 6.

    The patch size is 23.3% of the smaller image dimension and can cover up to 5.29% of the area of the image.

  7. 7.

    https://groups.csail.mit.edu/vision/datasets/ADE20K/.

  8. 8.

    https://vision.princeton.edu/projects/2010/SUN/.

References

  1. Bank, D., Greenfeld, D., Hyams, G.: Improved training for self training by confidence assessments. In: Arai, K., Kapoor, S., Bhatia, R. (eds.) SAI 2018. AISC, vol. 858, pp. 163–173. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-01174-1_13

    Chapter  Google Scholar 

  2. Bell, S., Upchurch, P., Snavely, N., Bala, K.: OpenSurfaces: a richly annotated catalog of surface appearance. ACM Trans. Graph. (TOG) 32(4), 1–17 (2013)

    Article  Google Scholar 

  3. Bell, S., Upchurch, P., Snavely, N., Bala, K.: Material recognition in the wild with the materials in context database. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 3479–3487 (2015)

    Google Scholar 

  4. Bokhovkin, A., Burnaev, E.: Boundary loss for remote sensing imagery semantic segmentation. In: Lu, H., Tang, H., Wang, Z. (eds.) ISNN 2019. LNCS, vol. 11555, pp. 388–401. Springer, Cham (2019). https://doi.org/10.1007/978-3-030-22808-8_38

    Chapter  Google Scholar 

  5. Chen, L.C., Zhu, Y., Papandreou, G., Schroff, F., Adam, H.: Encoder-decoder with atrous separable convolution for semantic image segmentation. In: Proceedings of the European Conference on Computer Vision (ECCV), pp. 801–818 (2018)

    Google Scholar 

  6. Chen, L., Tang, W., John, N.W., Wan, T.R., Zhang, J.J.: Context-aware mixed reality: a learning-based framework for semantic-level interaction. In: Computer Graphics Forum, vol. 39, pp. 484–496. Wiley Online Library (2020)

    Google Scholar 

  7. Chen, W., Jiang, Z., Wang, Z., Cui, K., Qian, X.: Collaborative global-local networks for memory-efficient segmentation of ultra-high resolution images. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 8924–8933 (2019)

    Google Scholar 

  8. Cheng, H., Gu, C., Wu, K.: Weakly-supervised semantic segmentation via self-training. In: Journal of Physics: Conference Series, vol. 1487, p. 012001. IOP Publishing (2020)

    Google Scholar 

  9. Delany, M., Bazley, E.: Acoustical properties of fibrous absorbent materials. Appl. Acoust. 3(2), 105–116 (1970)

    Article  Google Scholar 

  10. Deng, J., Dong, W., Socher, R., Li, L.J., Li, K., Fei-Fei, L.: Imagenet: a large-scale hierarchical image database. In: 2009 IEEE Conference on Computer Vision and Pattern Recognition, pp. 248–255. IEEE (2009)

    Google Scholar 

  11. Everingham, M., et al.: The 2005 PASCAL visual object classes challenge. In: Quiñonero-Candela, J., Dagan, I., Magnini, B., d’Alché-Buc, F. (eds.) MLCW 2005. LNCS (LNAI), vol. 3944, pp. 117–176. Springer, Heidelberg (2006). https://doi.org/10.1007/11736790_8

    Chapter  Google Scholar 

  12. Eversberg, L., Lambrecht, J.: Generating images with physics-based rendering for an industrial object detection task: realism versus domain randomization. Sensors 21(23), 7901 (2021)

    Article  Google Scholar 

  13. Farhadi, A., Endres, I., Hoiem, D., Forsyth, D.: Describing objects by their attributes. In: 2009 IEEE Conference on Computer Vision and Pattern Recognition, pp. 1778–1785. IEEE (2009)

    Google Scholar 

  14. Fleming, R.W.: Visual perception of materials and their properties. Vision. Res. 94, 62–75 (2014). https://doi.org/10.1016/j.visres.2013.11.004

    Article  Google Scholar 

  15. Ghiasi, G., Fowlkes, C.C.: Laplacian pyramid reconstruction and refinement for semantic segmentation. In: Leibe, B., Matas, J., Sebe, N., Welling, M. (eds.) ECCV 2016. LNCS, vol. 9907, pp. 519–534. Springer, Cham (2016). https://doi.org/10.1007/978-3-319-46487-9_32

    Chapter  Google Scholar 

  16. Goodfellow, I., Bengio, Y., Courville, A.: Deep Learning. MIT Press, Cambridge (2016)

    MATH  Google Scholar 

  17. 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)

    Google Scholar 

  18. Heng, Y., Wu, Y., Kim, H., Dasmahapatra, S.: Cam-segnet: a context-aware dense material segmentation network for sparsely labelled datasets. In: 17th International Conference on Computer Vision Theory and Applications (VISAPP), vol. 5, pp. 190–201 (2022)

    Google Scholar 

  19. Hodaň, T., et al.: Photorealistic image synthesis for object instance detection. In: 2019 IEEE International Conference on Image Processing (ICIP), pp. 66–70. IEEE (2019)

    Google Scholar 

  20. Iodice, S., Mikolajczyk, K.: Text attribute aggregation and visual feature decomposition for person search. In: BMVC (2020)

    Google Scholar 

  21. Kang, J., Fernandez-Beltran, R., Sun, X., Ni, J., Plaza, A.: Deep learning-based building footprint extraction with missing annotations. IEEE Geosci. Remote Sens. Lett. 19, 1–5 (2021)

    Google Scholar 

  22. Kim, H., Remaggi, L., Jackson, P.J., Hilton, A.: Immersive spatial audio reproduction for VR/AR using room acoustic modelling from 360 images. In: 2019 IEEE Conference on Virtual Reality and 3D User Interfaces (VR), pp. 120–126. IEEE (2019)

    Google Scholar 

  23. Krähenbühl, P., Koltun, V.: Parameter learning and convergent inference for dense random fields. In: International Conference on Machine Learning, pp. 513–521. PMLR (2013)

    Google Scholar 

  24. Le, T.H.N., Luu, K., Savvides, M.: Fast and robust self-training beard/moustache detection and segmentation. In: 2015 International Conference on Biometrics (ICB), pp. 507–512. IEEE (2015)

    Google Scholar 

  25. Lin, T.Y., Dollár, P., Girshick, R., He, K., Hariharan, B., Belongie, S.: Feature pyramid networks for object detection. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 2117–2125 (2017)

    Google Scholar 

  26. Lin, T.Y., Goyal, P., Girshick, R., He, K., Dollár, P.: Focal loss for dense object detection. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 2980–2988 (2017)

    Google Scholar 

  27. Liu, F., Ren, X., Zhang, Z., Sun, X., Zou, Y.: Rethinking skip connection with layer normalization. In: Proceedings of the 28th International Conference on Computational Linguistics, pp. 3586–3598 (2020)

    Google Scholar 

  28. Liu, Z., Chow, P., Xu, J., Jiang, J., Dou, Y., Zhou, J.: A uniform architecture design for accelerating 2D and 3D CNNs on FPGAs. Electronics 8(1), 65 (2019)

    Article  Google Scholar 

  29. Long, J., Shelhamer, E., Darrell, T.: Fully convolutional networks for semantic segmentation. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 3431–3440 (2015)

    Google Scholar 

  30. McDonagh, A., Lemley, J., Cassidy, R., Corcoran, P.: Synthesizing game audio using deep neural networks. In: 2018 IEEE Games, Entertainment, Media Conference (GEM), pp. 1–9. IEEE (2018)

    Google Scholar 

  31. Milioto, A., Vizzo, I., Behley, J., Stachniss, C.: Rangenet++: fast and accurate lidar semantic segmentation. In: 2019 IEEE/RSJ International Conference on Intelligent Robots and Systems (IROS), pp. 4213–4220. IEEE (2019)

    Google Scholar 

  32. Rahman, M.A., Wang, Y.: Optimizing intersection-over-union in deep neural networks for image segmentation. In: Bebis, G., et al. (eds.) ISVC 2016. LNCS, vol. 10072, pp. 234–244. Springer, Cham (2016). https://doi.org/10.1007/978-3-319-50835-1_22

    Chapter  Google Scholar 

  33. Ronneberger, O., Fischer, P., Brox, T.: U-Net: convolutional networks for biomedical image segmentation. In: Navab, N., Hornegger, J., Wells, W.M., Frangi, A.F. (eds.) MICCAI 2015. LNCS, vol. 9351, pp. 234–241. Springer, Cham (2015). https://doi.org/10.1007/978-3-319-24574-4_28

    Chapter  Google Scholar 

  34. Schwartz, G.: Visual Material Recognition. Drexel University (2018)

    Google Scholar 

  35. Schwartz, G., Nishino, K.: Visual material traits: recognizing per-pixel material context. In: Proceedings of the IEEE International Conference on Computer Vision Workshops, pp. 883–890 (2013)

    Google Scholar 

  36. Schwartz, G., Nishino, K.: Material recognition from local appearance in global context. In: Biology and Artificial Vision (Workshop held in conjunction with ECCV 2016) (2016)

    Google Scholar 

  37. Schwartz, G., Nishino, K.: Recognizing material properties from images. IEEE Trans. Pattern Anal. Mach. Intell. 42(8), 1981–1995 (2020). https://doi.org/10.1109/TPAMI.2019.2907850

    Article  Google Scholar 

  38. Shrivatsav, N., Nair, L., Chernova, S.: Tool substitution with shape and material reasoning using dual neural networks. arXiv preprint arXiv:1911.04521 (2019)

  39. Silberman, N., Hoiem, D., Kohli, P., Fergus, R.: Indoor segmentation and support inference from RGBD images. In: Fitzgibbon, A., Lazebnik, S., Perona, P., Sato, Y., Schmid, C. (eds.) ECCV 2012. LNCS, vol. 7576, pp. 746–760. Springer, Heidelberg (2012). https://doi.org/10.1007/978-3-642-33715-4_54

    Chapter  Google Scholar 

  40. Souly, N., Spampinato, C., Shah, M.: Semi supervised semantic segmentation using generative adversarial network. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 5688–5696 (2017)

    Google Scholar 

  41. Su, H., Jampani, V., Sun, D., Gallo, O., Learned-Miller, E., Kautz, J.: Pixel-adaptive convolutional neural networks. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition, pp. 11166–11175 (2019)

    Google Scholar 

  42. Sutton, C., McCallum, A.: An introduction to conditional random fields for relational learning. In: Introduction to Statistical Relational Learning, vol. 2, pp. 93–128 (2006)

    Google Scholar 

  43. Tang, Z., Bryan, N.J., Li, D., Langlois, T.R., Manocha, D.: Scene-aware audio rendering via deep acoustic analysis. IEEE Trans. Visual Comput. Graphics 26(5), 1991–2001 (2020)

    Article  Google Scholar 

  44. Tao, A., Sapra, K., Catanzaro, B.: Hierarchical multi-scale attention for semantic segmentation. arXiv preprint arXiv:2005.10821 (2020)

  45. Teichmann, M., Cipolla, R.: Convolutional crfs for semantic segmentation. In: BMVC (2019)

    Google Scholar 

  46. Wu, T., Lei, Z., Lin, B., Li, C., Qu, Y., Xie, Y.: Patch proposal network for fast semantic segmentation of high-resolution images. In: Proceedings of the AAAI Conference on Artificial Intelligence, vol. 34, pp. 12402–12409 (2020)

    Google Scholar 

  47. Xiao, J., Ehinger, K.A., Hays, J., Torralba, A., Oliva, A.: Sun database: exploring a large collection of scene categories. Int. J. Comput. Vision 119(1), 3–22 (2016)

    Article  MathSciNet  Google Scholar 

  48. Xiao, J., Hays, J., Ehinger, K.A., Oliva, A., Torralba, A.: Sun database: large-scale scene recognition from abbey to zoo. In: 2010 IEEE Computer Society Conference on Computer Vision and Pattern Recognition, pp. 3485–3492. IEEE (2010)

    Google Scholar 

  49. Yakubovskiy, P.: Segmentation models pytorch (2020). https://github.com/qubvel/segmentation_mo dels.pytorch

  50. Yu, C., Gao, C., Wang, J., Yu, G., Shen, C., Sang, N.: Bisenet v2: bilateral network with guided aggregation for real-time semantic segmentation. arXiv preprint arXiv:2004.02147 (2020)

  51. Zhang, H., Liao, Y., Yang, H., Yang, G., Zhang, L.: A local-global dual-stream network for building extraction from very-high-resolution remote sensing images. IEEE Trans. Neural Netw. Learn. Syst. 33(3), 1269–1283 (2020)

    Article  Google Scholar 

  52. Zhang, H., Shao, J., Salakhutdinov, R.: Deep neural networks with multi-branch architectures are intrinsically less non-convex. In: The 22nd International Conference on Artificial Intelligence and Statistics, pp. 1099–1109. PMLR (2019)

    Google Scholar 

  53. Zhao, C., Sun, L., Stolkin, R.: A fully end-to-end deep learning approach for real-time simultaneous 3D reconstruction and material recognition. In: 2017 18th International Conference on Advanced Robotics (ICAR), pp. 75–82. IEEE (2017)

    Google Scholar 

  54. Zhao, C., Sun, L., Stolkin, R.: Simultaneous material segmentation and 3D reconstruction in industrial scenarios. Front. Robot. AI 7, 52 (2020)

    Article  Google Scholar 

  55. Zhao, H., Shi, J., Qi, X., Wang, X., Jia, J.: Pyramid scene parsing network. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 2881–2890 (2017)

    Google Scholar 

  56. Zheng, S., et al.: Dense semantic image segmentation with objects and attributes. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 3214–3221 (2014)

    Google Scholar 

  57. Zheng, S., et al.: Conditional random fields as recurrent neural networks. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 1529–1537 (2015)

    Google Scholar 

  58. Zhou, B., Zhao, H., Puig, X., Fidler, S., Barriuso, A., Torralba, A.: Scene parsing through ADE20K dataset. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 633–641 (2017)

    Google Scholar 

  59. Zhu, X.J.: Semi-supervised learning literature survey (2005)

    Google Scholar 

  60. Zhu, X., et al.: Cylindrical and asymmetrical 3D convolution networks for lidar segmentation. arXiv preprint arXiv:2011.10033 (2020)

  61. Zoph, B., et al.: Rethinking pre-training and self-training. arXiv preprint arXiv:2006.06882 (2020)

Download references

Acknowledgements

This work was supported by the EPSRC Programme Grant Immersive Audio-Visual 3D Scene Reproduction Using a Single 360 Camera (EP/V03538X/1).

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to Hansung Kim .

Editor information

Editors and Affiliations

Rights and permissions

Reprints and permissions

Copyright information

© 2023 The Author(s), under exclusive license to Springer Nature Switzerland AG

About this paper

Check for updates. Verify currency and authenticity via CrossMark

Cite this paper

Heng, Y., Wu, Y., Dasmahapatra, S., Kim, H. (2023). Dense Material Segmentation with Context-Aware Network. In: de Sousa, A.A., et al. Computer Vision, Imaging and Computer Graphics Theory and Applications. VISIGRAPP 2022. Communications in Computer and Information Science, vol 1815. Springer, Cham. https://doi.org/10.1007/978-3-031-45725-8_4

Download citation

  • DOI: https://doi.org/10.1007/978-3-031-45725-8_4

  • Published:

  • Publisher Name: Springer, Cham

  • Print ISBN: 978-3-031-45724-1

  • Online ISBN: 978-3-031-45725-8

  • eBook Packages: Computer ScienceComputer Science (R0)

Publish with us

Policies and ethics