def run_calibration(args):
calibration_info = {
"model_name":
args.model_name,
"in_prototxt":
'./{}/deploy.prototxt'.format(args.model_name),
"in_caffemodel":
'./{}/{}.caffemodel'.format(args.model_name, args.model_name),
"iteration":
100,
"enable_memory_opt":
args.memory_opt,
"histogram_bin_num":
2048,
"math_lib_path":
'../../calibration_tool/lib/calibration_math.so'
}
print(calibration_info)
calib = Calibration(calibration_info)
calib.calc_tables()
calib.export_model('./{}/bmnet_{}_int8.caffemodel'.format(
args.model_name, args.model_name))
calib.export_calirabtion_pb2('./{}/bmnet_{}_calibration_table'.format(
args.model_name, args.model_name))
def run_calibration(arg):
calibration_info = {
"model_name": arg.model_name,
"in_prototxt": os.path.join(arg.model_path, 'deploy_2x.prototxt'),
"in_caffemodel": os.path.join(arg.model_path, 'espcn.caffemodel'),
"iteration": 100,
"enable_memory_opt": arg.memory_opt,
"histogram_bin_num": 2048,
"math_lib_path": "../../calibration_tool/build/calibration_math.so"
}
print(calibration_info)
calib = Calibration(calibration_info)
calib.calc_tables()
calib.export_model(
os.path.join(arg.model_path, 'bmnet_espcn_int8.caffemodel'))
calib.export_calirabtion_pb2(
os.path.join(arg.model_path, 'bmnet_espcn_calibration_table'))
def run_calibration(args):
calibration_info = {
"model_name": args.model_name,
"in_prototxt": os.path.join(args.model_path, 'deploy.prototxt'),
"in_caffemodel": os.path.join(args.model_path, 'custom.caffemodel'),
"iteration": 100,
"enable_memory_opt": args.memory_opt,
"enable_calibration_opt": 1,
"histogram_bin_num": 204800,
"math_lib_path": './lib/calibration_math.so',
"enable_concat_quantize": 0
}
print(calibration_info)
calib = Calibration(calibration_info)
calib.calc_tables()
calib.export_model('{}/bmnet_{}_int8.caffemodel'.format(
args.model_path, args.model_name))
calib.export_calirabtion_pb2('{}/bmnet_{}_calibration_table'.format(
args.model_path, args.model_name))
