Fiftyone Media Fields Example

thumbnails.py

import os
import shutil

import fiftyone as fo
import fiftyone.zoo as foz
import fiftyone.utils.image as foui

fo.core.dataset.delete_non_persistent_datasets()

exampleName = 'thumbnail-example'
exampleDir = "/tmp/quickstart"
dataset = foz.load_zoo_dataset("quickstart")

shutil.rmtree(exampleDir)

if fo.core.dataset.dataset_exists(exampleName):
    fo.core.dataset.delete_dataset(exampleName)

# # Export to a new place to avoid messing up zoo dataset
dataset.export(
    export_dir=exampleDir,
    dataset_type=fo.types.FiftyOneDataset,
)

dataset2 = fo.Dataset.from_dir(
    # name=exampleName,
    dataset_dir=exampleDir,
    dataset_type=fo.types.FiftyOneDataset,
)

# Generate some thumnbnails
for sample in dataset2.iter_samples(progress=True):
    root, ext = os.path.splitext(sample.filepath)
    sample["thumnbnail_path"] = root + "-thumbnail" + ext
    foui.transform_image(sample.filepath, sample.thumnbnail_path, size=(128, -1))
    sample.save()

# Here's a way that interacting with the `media_fields` might work
dataset2.app_config.media_fields.append("thumnbnail_path")
dataset2.app_config.grid_media_field = "thumnbnail_path"
dataset2.save()
# dataset2.compute_metadata(overwrite=True)

session = fo.launch_app(dataset2)
session.wait()