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
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