IDEAS home Printed from https://ideas.repec.org/a/gam/jmathe/v11y2023i8p1828-d1121679.html
   My bibliography  Save this article

A Cross-Modal Feature Fusion Model Based on ConvNeXt for RGB-D Semantic Segmentation

Author

Listed:
  • Xiaojiang Tang

    (College of Information and Electrical Engineering, China Agricultural University, Beijing 100083, China)

  • Baoxia Li

    (College of Information and Electrical Engineering, China Agricultural University, Beijing 100083, China)

  • Junwei Guo

    (College of Science, China Agricultural University, Beijing 100083, China)

  • Wenzhuo Chen

    (College of Information and Electrical Engineering, China Agricultural University, Beijing 100083, China)

  • Dan Zhang

    (College of Science, China Agricultural University, Beijing 100083, China)

  • Feng Huang

    (College of Science, China Agricultural University, Beijing 100083, China)

Abstract

Semantic segmentation, as the pixel level classification with dividing an image into multiple blocks based on the similarities and differences of categories (i.e., assigning each pixel in the image to a class label), is an important task in computer vision. Combining RGB and Depth information can improve the performance of semantic segmentation. However, there is still a problem of the way to deeply integrate RGB and Depth. In this paper, we propose a cross-modal feature fusion RGB-D semantic segmentation model based on ConvNeXt, which uses ConvNeXt as the skeleton network and embeds a cross-modal feature fusion module (CMFFM). The CMFFM designs feature channel-wise and spectral-wise fusion, which can realize the deeply feature fusion of RGB and Depth. The in-depth multi-modal feature fusion in multiple stages improves the performance of the model. Experiments are performed on the public dataset of SUN-RGBD, showing the best segmentation by our proposed model ConvNeXt-CMFFM with the highest mIoU score of 53.5% among the nine comparative models. The outstanding performance of ConvNeXt-CMFFM is also achieved on our self-built dataset of RICE-RGBD with the highest mIoU score and pixel accuracy among the three comparative datasets. The ablation experiment on our rice dataset shows that compared with ConvNeXt (without CMFFM), the mIoU score of ConvNext-CMFFM is increased from 71.5% to 74.8% and its pixel accuracy is increased from 86.2% to 88.3%, indicating the effectiveness of the added feature fusion module in improving segmentation performance. This study shows the feasibility of the practical application of the proposed model in agriculture.

Suggested Citation

  • Xiaojiang Tang & Baoxia Li & Junwei Guo & Wenzhuo Chen & Dan Zhang & Feng Huang, 2023. "A Cross-Modal Feature Fusion Model Based on ConvNeXt for RGB-D Semantic Segmentation," Mathematics, MDPI, vol. 11(8), pages 1-13, April.
  • Handle: RePEc:gam:jmathe:v:11:y:2023:i:8:p:1828-:d:1121679
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2227-7390/11/8/1828/pdf
    Download Restriction: no

    File URL: https://www.mdpi.com/2227-7390/11/8/1828/
    Download Restriction: no
    ---><---

    References listed on IDEAS

    as
    1. Juan Duarte & Guillermo Martínez-Flórez & Diego Ignacio Gallardo & Osvaldo Venegas & Héctor W. Gómez, 2023. "A Bimodal Extension of the Epsilon-Skew-Normal Model," Mathematics, MDPI, vol. 11(3), pages 1-18, January.
    Full references (including those not matched with items on IDEAS)

    Most related items

    These are the items that most often cite the same works as this one and are cited by the same works as this one.

      Corrections

      All material on this site has been provided by the respective publishers and authors. You can help correct errors and omissions. When requesting a correction, please mention this item's handle: RePEc:gam:jmathe:v:11:y:2023:i:8:p:1828-:d:1121679. See general information about how to correct material in RePEc.

      If you have authored this item and are not yet registered with RePEc, we encourage you to do it here. This allows to link your profile to this item. It also allows you to accept potential citations to this item that we are uncertain about.

      If CitEc recognized a bibliographic reference but did not link an item in RePEc to it, you can help with this form .

      If you know of missing items citing this one, you can help us creating those links by adding the relevant references in the same way as above, for each refering item. If you are a registered author of this item, you may also want to check the "citations" tab in your RePEc Author Service profile, as there may be some citations waiting for confirmation.

      For technical questions regarding this item, or to correct its authors, title, abstract, bibliographic or download information, contact: MDPI Indexing Manager (email available below). General contact details of provider: https://www.mdpi.com .

      Please note that corrections may take a couple of weeks to filter through the various RePEc services.

      IDEAS is a RePEc service. RePEc uses bibliographic data supplied by the respective publishers.