https://github.com/nikhilroxtomar/image-segmentation-using-unetr

Human Hair Segmentation using the UNETR (U-Net Transformers) in TensorFlow
https://github.com/nikhilroxtomar/image-segmentation-using-unetr

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

Last synced: 5 months ago
JSON representation

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

• Host: GitHub
• URL: https://github.com/nikhilroxtomar/image-segmentation-using-unetr
• Owner: nikhilroxtomar
• License: apache-2.0
• Created: 2024-01-28T16:54:10.000Z (over 1 year ago)
• Default Branch: main
• Last Pushed: 2024-02-02T12:43:11.000Z (over 1 year ago)
• Last Synced: 2024-11-16T06:28:02.644Z (7 months ago)
• Topics: image-segmentation-tensorflow, unet-image-segmentation, unetr, unetr-segmentation
• Language: Python
• Homepage:
• Size: 329 KB
• Stars: 4
• Watchers: 2
• Forks: 1
• Open Issues: 1
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

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.](figures/unetr_architecture.png) |

| :--: |

| *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`. 
 


| ![](results/92.jpg) |

| :--: |

| ![](results/595.jpg) |

| ![](results/626.jpg) |

## 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