Computer vision
Beta
This feature is in Beta.
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.
Before running these notebooks, see the Best practices checklist.
Object detection using DETR
This notebook demonstrates how to train an object detection model using a Hugging Face example on one A10 GPU. The example uses the Detection Transformer (DETR) model architecture for object detection tasks.
Notebook
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.