File size: 1,062 Bytes
747451d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | # /*---------------------------------------------------------------------------------------------
# * Copyright (c) 2025 STMicroelectronics.
# * All rights reserved.
# *
# * This software is licensed under terms that can be found in the LICENSE file in
# * the root directory of this software component.
# * If no LICENSE file comes with this software, it is provided AS-IS.
# *--------------------------------------------------------------------------------------------*/
from common.registries.quantizer_registry import QUANTIZER_WRAPPER_REGISTRY
from object_detection.tf.src.quantization.onnx_quantizer import OnnxPTQQuantizer
__all__ = ['OnnxPTQQuantizer']
# Register the ONNX PTQ Quantizer class from another folder
QUANTIZER_WRAPPER_REGISTRY.register(
quantizer_name="onnx_quantizer",
framework="tf",
use_case="object_detection"
)(OnnxPTQQuantizer)
QUANTIZER_WRAPPER_REGISTRY.register(
quantizer_name="onnx_quantizer",
framework="torch",
use_case="object_detection"
)(OnnxPTQQuantizer)
|