probly.datasets.torch.Treeversity1

class probly.datasets.torch.Treeversity1(root: Path | str, transform: Callable[..., Any] | None = None, *, first_order: bool = True)

Bases: DCICDataset

Implementation of the Treeversity#1 dataset.

The dataset can be found at https://zenodo.org/records/7180818.

Initialize an instance of the Treeversity#1 dataset class.

Parameters:

• root – Root directory of the dataset.

• transform – Optional transform to apply to the data.

• first_order – Whether to use first order data or class labels. Defaults to True.

data

List of images.

image_labels

Dictionary of image labels grouped by image.

image_paths

List of image paths.

label_mappings

Mapping of labels to indices.

num_classes

Number of classes.

root

Root directory of the dataset.

targets

List of labels.

transform

Transform to apply to the data.