objdettargettransformer

objdettargettransformer ¶

Module for the input and target transformers used for object detection

Classes¶

ObjDetTargetTransformer ¶

ObjDetTargetTransformer(anchor_generator, anchor_encoder)

Bases: NetTargetTransformer

Target transformer for object detection

Source code in niceml/mlcomponents/targettransformer/objdettargettransformer.py

def __init__(
    self, anchor_generator: AnchorGenerator, anchor_encoder: AnchorEncoder
):
    self.anchor_encoder = anchor_encoder
    self.anchor_generator = anchor_generator
    self.anchors = None