Skip to content

collection of simple and reusable Python scripts for image segmentation tasks — including mask processing, RGB color extraction, visualization, metrics, and more.

Notifications You must be signed in to change notification settings

nikhilroxtomar/segmentation-utils

Repository files navigation

🧠 Segmentation Utils

A collection of simple and reusable Python scripts for image segmentation tasks — including mask processing, RGB color extraction, visualization, metrics, and more.


📌 Features

  • ✅ Extract unique RGB values from segmentation masks
  • 🎨 Convert RGB masks to label indices and vice versa
  • 📊 Evaluate segmentation with metrics like IoU and Dice Score
  • 🖼️ Overlay masks on images for visualization
  • ⚙️ Easily customizable for your dataset and model output

Requirements

  • Python 3.7+
  • OpenCV
  • NumPy
  • tqdm
  • PIL (Pillow)

🤝 Contributions

Pull requests, suggestions, and bug reports are welcome!
If you have a useful segmentation utility, feel free to submit a PR.

Author

Created by Nikhil Tomar
For more tutorials and guides, follow the repo and stay updated!

About

collection of simple and reusable Python scripts for image segmentation tasks — including mask processing, RGB color extraction, visualization, metrics, and more.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages