qualia_codegen_plugin_snn.graph.layers.TObjectDetectionPostProcessLayer module

class qualia_codegen_plugin_snn.graph.layers.TObjectDetectionPostProcessLayer.TObjectDetectionPostProcessLayer(input_shape: qualia_codegen_core.typing.Shapes, output_shape: qualia_codegen_core.typing.Shapes, output_dtype: qualia_codegen_core.typing.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]

input_shape: Shapes

output_shape: Shapes

output_dtype: DTypes

name: str