/demo-image-classification

Primary LanguageJupyter NotebookApache License 2.0Apache-2.0

Image Classification Using Distributed Training

This example is using TensorFlow, Horovod, and Nuclio demonstrating end to end solution for image classification, it consists of 4 MLRun and Nuclio functions:

  1. import an image archive from S3 to the cluster file system
  2. Tag the images based on their name structure
  3. Distrubuted training using TF, Keras and Horovod
  4. Automated deployment of Nuclio model serving function (form Notebook and from Dockerfile)



The Example also demonstrate an automated pipeline using MLRun and KubeFlow pipelines

Notebooks & Code