Python save_tflite示例

示例#1

def test_softmax():
    ncc.clear()
    ncc.save_tflite(module)
    init_values()
    ncc.compile(['--inference-type', 'float'])

    ncc.infer(['--dataset-format', 'raw'])
    ncc.close_to('test', 0)

示例#2

def test_simple_conv2d():
    ncc.clear()
    ncc.save_tflite(module)
    init_values()
    ncc.compile(['--inference-type', 'float', '-t', 'cpu'])

    ncc.infer(['--dataset-format', 'raw'])
    ncc.close_to('test', 1e-4)

示例#3

def test_simple_matmul_quant():
	ncc.clear()
	ncc.save_tflite(module)
	init_values()
	ncc.compile(['--inference-type', 'uint8', '-t', 'cpu',
	 '--dataset', ncc.input_dir + '/test.bin', '--dataset-format', 'raw',
	 '--input-type', 'float'])

	ncc.infer(['--dataset-format', 'raw'])
	ncc.close_to('test', 1e-3)

示例#4

def test_simple_conv2d_k210():
    ncc.clear()
    ncc.save_tflite(module)
    init_values()
    ncc.compile([
        '--inference-type', 'uint8', '-t', 'k210', '--dataset',
        ncc.input_dir + '/test.bin', '--dataset-format', 'raw', '--input-type',
        'float'
    ])

    ncc.infer(['--dataset-format', 'raw'])
    ncc.close_to('test', 0.4)