nikhilroxtomar / Image-Segmentation-using-UNETR

Human Hair Segmentation using the UNETR (U-Net Transformers) in TensorFlow

Apache License 2.0

4 stars 1 forks source link

image-segmentation-tensorflow unet-image-segmentation unetr unetr-segmentation

readme

Image-Segmentation-using-UNETR

This GitHub repository demonstrates the utilization of UNETR for Human Hair Segmentation using TensorFlow.

Architecture


The block diagram of the Original UNETR model.

Dataset

Download the dataset: Soon.

Results

The sequence in the images below is `Input Image`, `Ground Truth` and `Prediction`.

How to improve

Train on more epochs.
Increase the input image resolution.
Apply data augmentation.

Contact

For more follow me on: