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

class cerebras.modelzoo.data.multimodal.datasets.OpenImages.OpenImagesv4(data_dir, split, *args)[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`

static display_sample(features_dict)[source]#: Plot and display features_dict

cerebras.modelzoo.data.multimodal.datasets.OpenImages

cerebras.modelzoo.data.multimodal.datasets.VQAv2