Skip to content

Image Segmentation with TensorFlow, A simple pipeline for training and testing image segmentation models using TFRecords.

Notifications You must be signed in to change notification settings

DevEooo/Image-Segmentation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

2 Commits
Β 
Β 
Β 
Β 

Repository files navigation

Image Segmentation with TensorFlow

This project demonstrates how to build an image segmentation pipeline using TensorFlow and TFRecords.
It covers dataset loading, preprocessing, batching, and visualization of input images, labels, and predictions.

πŸ“Œ Features

  • Load training and validation datasets from TFRecords.
  • Parse and decode image/label data into TensorFlow tensors.
  • Preprocess images (reshape, normalize, cast).
  • Create optimized tf.data.Dataset pipelines for training and testing.
  • Display input images, ground-truth labels, and predictions with Matplotlib.
  • Ready-to-extend into a full segmentation model (U-Net, FCN, etc.).

About

Image Segmentation with TensorFlow, A simple pipeline for training and testing image segmentation models using TFRecords.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages