14 / 2021-06-14 16:49:15
An Improved U-Net Network and Its Application in Gear Pitting Measurement
终稿
四军 王 / Chongqing university
毅 秦 / 重庆大学
德君 奚 / Chongqing university
Aiming at the problems of low segmentation accuracy of small targets, high computational complexity in U-Net, a new U-Net network based on dilated convolution and reconstructed sampling units (DSU-Net) is proposed. In DSU-Net, in order to increase the receptive field of image feature extraction and fuse multi-scale information, dilated convolutional layers with different dilation rates are designed; in view of the shortcoming of losing a large amount of semantic information during the pooling process, sampling units that combine pooling and convolution are constructed, and depthwise separable convolution is used for feature extraction, thereby enhancing the feature extraction capability of neural network and reducing the computational cost. The experimental results of Gear Pitting dataset show that DSU-Net has better segmentation performance than U-Net, ResU-Net and R2U-Net on the three metrics of IoU, Dice Coeff and F1 Score. The proposed method can calculate the gear pitting area ratio more accurately, so as to solve the problem of efficiently and accurately detecting gear failure in the gear contact fatigue test.Aiming at the problems of low segmentation accuracy of small targets, high computational complexity in U-Net, a new U-Net network based on dilated convolution and reconstructed sampling units (DSU-Net) is proposed. In DSU-Net, in order to increase the receptive field of image feature extraction and fuse multi-scale information, dilated convolutional layers with different dilation rates are designed; in view of the shortcoming of losing a large amount of semantic information during the pooling process, sampling units that combine pooling and convolution are constructed, and depthwise separable convolution is used for feature extraction, thereby enhancing the feature extraction capability of neural network and reducing the computational cost. The experimental results of Gear Pitting dataset show that DSU-Net has better segmentation performance than U-Net, ResU-Net and R2U-Net on the three metrics of IoU, Dice Coeff and F1 Score. The proposed method can calculate the gear pitting area ratio more accurately, so as to solve the problem of efficiently and accurately detecting gear failure in the gear contact fatigue test.
重要日期
  • 会议日期

    10月21日

    2021

    10月23日

    2021

  • 10月26日 2021

    注册截止日期

主办单位
Southeast University, China
移动端
在手机上打开
小程序
打开微信小程序
客服
扫码或点此咨询