cerebras.modelzoo.data.multimodal.datasets.OpenImages.OpenImagesv4#

class cerebras.modelzoo.data.multimodal.datasets.OpenImages.OpenImagesv4[source]#

Bases: cerebras.modelzoo.data.multimodal.datasets.BaseDataset.BaseDataset

This class builds an OpenImages dataset based on version v4 and all metadata info

Methods

classid_to_classintid

classid_to_label

classintid_to_classid

classintid_to_label

display_sample

Plot and display features_dict

index_to_sample_path

label_to_classid

label_to_classintid

process_bboxes

process_image_labels

sample_path_to_index

Attributes

label_id_range

num_labels

__init__(data_dir, split, *args)[source]#
static display_sample(features_dict)[source]#

Plot and display features_dict

previous

cerebras.modelzoo.data.multimodal.datasets.OpenImages

next

cerebras.modelzoo.data.multimodal.datasets.VQAv2