lafith/Mobile-UNet

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