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(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)
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))