Skip to main content

Computer vision

Beta

This feature is in Beta. Workspace admins can control access to this feature from the Previews page. See Manage 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.

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 GPU compute for image classification.

Object detection using RetinaNet

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