arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.02656eess.IVcs.CV

GPU-Net: Lightweight U-Net with more diverse features

Tsinghua University(清华大学);University of Southern California(南加州大学);Hebei University of Engineering(河北工程大学)

Heng Yu, Di Fan, Weihu Song

英文摘要:

Image segmentation is an important task in the medical image field and many convolutional neural networks (CNNs) based methods have been proposed, among which U-Net and its variants show promising performance. In this paper, we propose GP-module and GPU-Net based on U-Net, which can learn more diverse features by introducing Ghost module and atrous spatial pyramid pooling (ASPP). Our method achieves better performance with more than 4 times fewer parameters and 2 times fewer FLOPs, which provides a new potential direction for future research. Our plug-and-play module can also be applied to existing segmentation methods to further improve their performance.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑