CNN_ImageSegmentation

Fully Convolutional Image Segmentation based on ResNet50.

Model trained on detecting dice

Content

Training.ipynb Jupyter Notebook for training
cnn_image_segmentation/resnet_segmentation_model.py Module containe the code to create the model

Dependencies

Tensorflow 2.0
Python 3

Setup

cd CNN_ImageSegmentation python setup.py develop

References

Deep Residual Learning for Image Recognition
Fully Convolutional Networks for Semantic Segmentation