Skip to content

There are three levels in image analysis, Classification, Detection and Segmentation. Image segmentation is the division of an image into regions or categories. The goal of segmentation is to simplify and/or change the representation of an image into something that is more meaningful and easier to analyze.

master
Switch branches/tags
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 

COCO Dataset 2018 Stuff Segmentation Challenge

Internship project in Bennett University under Leadinginadi.ai

Getting Started

Problem Statement :-
To perform Semantic Segmentation of Stuff classes.The COCO Stuff Segmentation Task is designed to push the state of the art in semantic segmentation of stuff classes.

Prerequisites and installing

pip install tensorflow-gpu
pip install tqdm
pip install keras
pip install keras-segmentation

Dataset format

You need to make two folders

Images Folder - For all the training images
Annotations Folder - For the corresponding ground truth segmentation images
The filenames of the annotation images should be same as the filenames of the RGB images.

Usage via command line

Visualizing the prepared data

python -m keras_segmentation verify_dataset \
 --images_path="dataset1/images_prepped_train/" \
 --segs_path="dataset1/annotations_prepped_train/"  \
 --n_classes=50
python -m keras_segmentation visualize_dataset \
 --images_path="dataset1/images_prepped_train/" \
 --segs_path="dataset1/annotations_prepped_train/"  \
 --n_classes=50

Training the Model

python -m keras_segmentation train \
--checkpoints_path="path_to_checkpoints" \
--train_images="dataset1/images_prepped_train/" \
--train_annotations="dataset1/annotations_prepped_train/" \
--val_images="dataset1/images_prepped_test/" \
--val_annotations="dataset1/annotations_prepped_test/" \
--n_classes=300 \
--input_height=320 \
--input_width=640 \
--model_name="pspnet"

Getting the predictions

python -m keras_segmentation predict \
 --checkpoints_path="path_to_checkpoints" \
 --input_path="dataset1/images_prepped_test/" \
 --output_path="path_to_predictions"

References

  1. https://github.com/divamgupta/image-segmentation-keras
  2. https://github.com/GeorgeSeif/Semantic-Segmentation-Suite
  3. https://github.com/aurora95/Keras-FCN
  4. http://cocodataset.org/#stuff-2018
  5. https://arxiv.org/abs/1612.03716

About

There are three levels in image analysis, Classification, Detection and Segmentation. Image segmentation is the division of an image into regions or categories. The goal of segmentation is to simplify and/or change the representation of an image into something that is more meaningful and easier to analyze.

Resources

License

Releases

No releases published

Packages

No packages published