cerebras.modelzoo.data.multimodal.datasets.features.ObjectDetectionFeaturesDict#
- class cerebras.modelzoo.data.multimodal.datasets.features.ObjectDetectionFeaturesDict[source]#
Bases:
object
ImageID: Name of image
Methods
compare
Attributes
Bboxes
Image
Objects
ImagePath
ImageID
- __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 #