Data Types

Data Types in W&B are classes that wrap media and structured data for logging to runs. They include visualization components in the W&B UI and handle data serialization, storage, and retrieval.

Available Data Types

Data Type Description
Image Log images with support for masks, bounding boxes, and segmentation.
Video Track video data for model outputs or dataset samples.
Audio Log audio samples for audio processing tasks.
Table Create tables that can contain mixed media types.
Plotly Log Plotly charts for data visualization.
Html Embed custom HTML content.
Object3D Visualize 3D point clouds and meshes.
Molecule Log molecular structures for computational chemistry.

Examples

This example uses an Image:

import wandb
import matplotlib.pyplot as plt

# Generate an image for demonstration purposes
path_to_img = "/path/to/cafe.png"
im = plt.imread(path_to_img)

# Initialize a new run
with wandb.init(project="awesome-project") as run:

    # Log the image
    run.log({"img": [wandb.Image(im, caption="Cafe")]})

This example uses a Table to log a table with mixed text and labels:

import wandb

# Initialize a new run
with wandb.init(project="visualize-predictions", name="tables") as run:

    # Create tabular data, using a list of lists
    data = [["Cat", "1", "1"],["Dog", "0", "-1"]]
    run.log({"Table 1": wandb.Table(data=data, columns=["Text", "Predicted Label", "True Label"])})

    # Create tabular data, using `wandb.Table.add_data()` method
    table = wandb.Table(columns=["Text", "Predicted Label", "True Label"])
    table.add_data("Cat", "1", "1")
    table.add_data("Dog", "0", "-1")
    run.log({"Table 2": table})