cerebras.modelzoo.data.multimodal.datasets.features.ObjectDetectionFeaturesDict#
- class cerebras.modelzoo.data.multimodal.datasets.features.ObjectDetectionFeaturesDict[source]#
Bases:
objectImageID: Name of image
Methods
compareAttributes
BboxesImageObjectsImagePathImageID- __init__(ImagePath: str, ImageID: str, Image: Optional[torch.Tensor] = None, Bboxes: Optional[List[cerebras.modelzoo.data.multimodal.datasets.features.Bbox]] = None, Objects: Optional[List[cerebras.modelzoo.data.multimodal.datasets.features.ImageLabels]] = None) None#