- 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](https://github.com/PINTO0309/onnx2tf/files/10884363/cf_fus.onnx.zip)
![image](https://user-images.githubusercontent.com/33194443/222779034-78192d45-5992-4410-b2fd-a929e9aeaf02.png)
python
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](https://github.com/PINTO0309/onnx2tf/issues/222)
What's Changed
* Improved error judgment regarding calibration data for INT8 quantization by PINTO0309 in https://github.com/PINTO0309/onnx2tf/pull/224
**Full Changelog**: https://github.com/PINTO0309/onnx2tf/compare/1.7.13...1.7.14