qualia_codegen_plugin_snn.graph.layers.TObjectDetectionPostProcessLayer module

class qualia_codegen_plugin_snn.graph.layers.TObjectDetectionPostProcessLayer.TObjectDetectionPostProcessLayer(input_shape: 'Shapes', output_shape: 'Shapes', output_dtype: 'DTypes', name: 'str', num_classes: 'int', num_fms: 'int', image_shape: 'tuple[int, ...]', score_threshold: 'float', nms_threshold: 'float', topk_candidates: 'int', detections_per_image: 'int', box_coder_weights: 'list[float]', anchors: 'NDArrayFloatOrInt')[source]

Bases: TBaseLayer

num_classes: int

num_fms: int

image_shape: tuple[int, ...]

score_threshold: float

nms_threshold: float

topk_candidates: int

detections_per_image: int

box_coder_weights: list[float]

anchors: NDArrayFloatOrInt

property weights: OrderedDict[str, NDArrayFloatOrInt]