Share via


Computer vision

Important

This feature is in Beta. Workspace admins can control access to this feature from the Previews page. See Manage Azure Databricks previews.

This page provides notebook examples for computer vision tasks using Serverless GPU compute. These examples demonstrate how to train and fine-tune models for various computer vision applications.

Image classification using convolutional neural network

The following notebook provides a simple example of how to train a 2-D convolution neural network on Serverless GPU compute for image classification.

PyTorch image classification

Get notebook

Object detection using RetinaNet

The following notebook demonstrates how to train an object detection model using RetinaNet on Serverless GPU compute.

RetinaNet image detection

Get notebook