/dl-demo

Deep Learning computer vision demo

Primary LanguageJupyter Notebook

Deep Learning Demo

Mnist

Tensorflow-based demo for recognition of hand-written digits. Training is performed on the MNIST dataset.

mnist/mnist_slim.ipynb -> Build, train and test the model

Cat-Dog

Tensorflow-based demo for binary image classification [Cat | Dog]. Training is performed on the ImageNet subset.

Tensorflow_train_model.ipynb -> Build and train the model

Tensorflow_freeze_graph.ipynb -> Freeze Tensorflow graph (prepare for the inference)

Tensorflow_inference.ipynb -> Inference