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:

• YouTube
• Facebook
• Twitter
• Instagram
• Telegram