def run_calibration(in_proto,
in_caffemodel,
out_table,
out_caffemodel,
thresholds=None):
calibration_info = {
"model_name": 'tune',
"in_prototxt": in_proto,
"in_caffemodel": in_caffemodel,
"out_prototxt": './deploy_out.prototxt',
"iteration": 1,
"enable_memory_opt": 0,
"math_lib_path": '../../calibration_tool/lib/calibration_math.so'
}
print(calibration_info)
calib = Calibration(calibration_info)
calib.tune_tables(thresholds)
calib.export_model(out_caffemodel)
calib.export_calirabtion_pb2(out_table)
