learnopencv/Medical-Image-Segmentation-Using-HuggingFace-&-PyTorch

Medical Image Segmentation Using 🤗 HuggingFace & PyTorch

This repository contains the files related to the LearnOpenCV blog post: Medical Image Segmentation Using 🤗 HuggingFace & PyTorch

Medical image segmentation is an innovative process that enables surgeons to have a virtual "x-ray vision." It is a highly valuable tool in healthcare, providing non-invasive diagnostics and in-depth analysis. With this in mind, in this post, we will explore the UW-Madison GI Tract Image Segmentation Kaggle challenge dataset. As part of this project, we will utilize PyTorch along with PyTorch-Lightning. We will use 🤗 HuggingFace transformers to load and fine-tune the Segformer transformer-based model on the medical segmentation dataset. Finally, we will create a Gradio app for image inference and deploy it on HuggingFace spaces.

---

You can use the Gradio inference demo app here --> Medical Image Segmentation Gradio App

All the files required can be accessed from here --> Gradio App Files

---

---

AI Courses by OpenCV

Want to become an expert in AI? AI Courses by OpenCV is a great place to start.