Skip to main content

Vision Tasks

Train models for image classification, object detection, and vision-language tasks.

Image Classification

Quick Start

Parameters

Data Format

Folder structure:
Or CSV with paths:

Example: Custom Classifier

Image Regression

For predicting continuous values from images.

Quick Start

Parameters

Example: Age Prediction

Object Detection

Quick Start

Data Format

COCO-style format:

Parameters

Example: Custom Detection

Vision-Language Models (VLM)

Train models that understand both images and text.

Quick Start

Data Format

Parameters

Example: Image Captioning

Common Models

Image Classification

Object Detection

Vision-Language

GPU Memory Tips

  • Use smaller batch sizes for large images
  • Enable gradient checkpointing for VLMs
  • Use LoRA for VLM training:

Next Steps

Text Tasks

Text classification and NER

Tabular Data

Train on structured data