1.7.14
- Improved error judgment regarding calibration data for INT8 quantization.
- Models with multiple input OPs and non-rgb-image input OPs force automatic calibration to be aborted.
- e.g. cf_fus.onnx.zip
if model_input.dtype != tf.float32 \ or len(model_input.shape) != 4 \ or model_input.shape[-1] != 3: print( f'{Color.RED}ERROR:{Color.RESET} ' + f'For models that have multiple input OPs and need to perform INT8 quantization calibration '+ f'using non-rgb-image input tensors, specify the calibration data with '+ f'--quant_calib_input_op_name_np_data_path. '+ f'model_input[n].shape: {model_input.shape}' ) sys.exit(1)
- [CenterFusion] Model Full Interger Quantize problem #222
What's Changed
- Improved error judgment regarding calibration data for INT8 quantization by @PINTO0309 in #224
Full Changelog: 1.7.13...1.7.14