zilbermanor / demo-image-classification

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

• All-in-one: Import, tag, launch training, deploy serving
• Training function code
• Serving function development and testing
• Auto generation of KubeFlow pipelines workflow
• Building serving function using Dockerfile
  • function code
  • Dockerfile