Skip to content

Latest commit

 

History

History
33 lines (25 loc) · 996 Bytes

README.md

File metadata and controls

33 lines (25 loc) · 996 Bytes

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.
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: