Computer vision

Important

This feature is in Public Preview.

These notebooks train computer vision models on AI Runtime. They cover image classification with a convolutional neural network and object detection with RetinaNet.

Tutorial	Description
Image classification using convolutional neural network	This notebook provides a simple example of how to train a 2-D convolution neural network on serverless GPUs for image classification.
Object detection using RetinaNet	This notebook demonstrates how to train an object detection model using RetinaNet on serverless GPU.
Object detection using YOLO11n	This notebook demonstrates how to train a YOLO11n object detection model on the COCO128 dataset using serverless GPU, with MLflow tracking and Model Serving deployment.