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

A Lightweight YOLOv5-Based Model with Feature Fusion and Dilation Convolution for Image Segmentation

Author

Listed:
  • Linwei Chen

    (College of Information Engineering, China Jiliang University, Hangzhou 310018, China)

  • Jingjing Yang

    (School of Information Science and Engineering, Hebei North University, Zhangjiakou 075000, China)

Abstract

Image segmentation has played an essential role in computer vision. The target detection model represented by YOLOv5 is widely used in image segmentation. However, YOLOv5 has performance bottlenecks such as object scale variation, object occlusion, computational volume, and speed when processing complex images. To solve these problems, an enhanced algorithm based on YOLOv5 is proposed. MobileViT is used as the backbone network of the YOLOv5 algorithm, and feature fusion and dilated convolution are added to the model. This method is validated on the COCO and PASCAL-VOC datasets. Experimental results show that it significantly reduces the processing time and achieves high segmentation quality with an accuracy of 95.32% on COCO and 96.02% on PASCAL-VOC. The improved model is 116 M, 52 M, and 76 M, smaller than U-Net, SegNet, and Mask R-CNN, respectively. This paper provides a new idea and method with which to solve the problems in the field of image segmentation, and the method has strong practicality and generalization value.

Suggested Citation

  • Linwei Chen & Jingjing Yang, 2023. "A Lightweight YOLOv5-Based Model with Feature Fusion and Dilation Convolution for Image Segmentation," Mathematics, MDPI, vol. 11(16), pages 1-17, August.
    • Handle: RePEc:gam:jmathe:v:11:y:2023:i:16:p:3538-:d:1218373
    as

    Download full text from publisher

    File URL: https://www.mdpi.com/2227-7390/11/16/3538/pdf
    Download Restriction: no
    File URL: https://www.mdpi.com/2227-7390/11/16/3538/
    Download Restriction: no
    ---><---

    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:16:p:3538-:d:1218373. 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.

    We have no bibliographic references for this item. You can help adding them by using 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.