def test_simple(): ncc.clear() ncc.copy_tflite(dir_prefix + '/model/20classes_yolo.tflite') init_values() ncc.compile( ['--inference-type', 'float', '--max-allocator-solve-secs', '0']) ncc.infer(['--dataset-format', 'raw']) ncc.close_to('test', 1e-6)
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)
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)
def test_simple_k210(): ncc.clear() ncc.copy_tflite('../examples/20classes_yolo/model/20classes_yolo.tflite') init_values() ncc.compile([ '--inference-type', 'uint8', '-t', 'k210', '--dataset', ncc.input_dir + '/test.bin', '--dataset-format', 'raw', '--input-type', 'float', '--max-allocator-solve-secs', '0' ]) ncc.infer(['--dataset-format', 'raw']) ncc.close_to('test', 1.3)