Skip to content

imageinputtransformer

imageinputtransformer

Module for ImageInputTransformer

Classes

ImageInputTransformer

ImageInputTransformer(image_attr_name='image')

Bases: NetInputTransformer

Input transformer for object detection

Source code in niceml/mlcomponents/targettransformer/imageinputtransformer.py
def __init__(self, image_attr_name: str = "image"):
    self.image_attr_name = image_attr_name