This GitHub repository demonstrates the utilization of UNETR for Human Hair Segmentation using TensorFlow.
 |
|---|
| The block diagram of the Original UNETR model. |
Download the dataset: Soon.
The sequence in the images below is Input Image, Ground Truth and Prediction.
 |
|---|
 |
 |
- Train on more epochs.
- Increase the input image resolution.
- Apply data augmentation.
For more follow me on: