Mobile-UNet icon indicating copy to clipboard operation
Mobile-UNet copied to clipboard

PyTorch implementation of Mobile-UNet.

Mobile UNet

PyTorch implementation of Mobile UNet. It is a modification of UNet with Inverted residual blocks and Depthwise seperable convolution. Mobile-UNet is optimized for attaining real-time image segmentation.

References

  1. Mobile-Unet: An efficient convolutional neural network for fabric defect detection
  2. Real-Time Hair Segmentation Using Mobile-Unet