Esempi in Python per save_expect_array

Esempio n. 1

File: test_conv2d.py Progetto: observerai/nncase

def init_values():
    input = np.arange(1, 16 + 1, dtype=np.float32).reshape([1, 1, 4, 4])
    ncc.save_input_array('test', input)
    ncc.save_expect_array(
        'test',
        np.transpose(ncc.run_tflite(np.transpose(input, [0, 2, 3, 1])),
                     [0, 3, 1, 2]))

Esempio n. 2

File: test_20classes_yolo.py Progetto: osatman/nncase

def init_values():
    input = [
        normalize(
            plt.imread(dir_prefix + '/k210/kpu_20classes_example/dog.bmp'))
    ]
    expect = ncc.run_tflite(input)
    expect = np.transpose(expect, [0, 3, 1, 2])
    ncc.save_expect_array('test', expect)
    input = np.transpose(input, [0, 3, 1, 2])
    ncc.save_input_array('test', input)

Esempio n. 3

File: test_add.py Progetto: zhen8838/nncase

def test_add(input):
    ncc.clear()

    ncc.save_input_array('test', input)
    onnx_importer.utils.save(module, torch.from_numpy(input))
    ncc.save_expect_array('test', onnx_importer.utils.run(input))

    onnx_importer.utils.compile(['--inference-type', 'float'])

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

Esempio n. 4

def test_matmul_quant(input):
    ncc.clear()

    ncc.save_input_array('test', input)
    onnx_importer.utils.save(module, torch.from_numpy(input))
    ncc.save_expect_array('test', onnx_importer.utils.run(input))

    onnx_importer.utils.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', 0.005)

Esempio n. 5

def test_clip_k210_onnx_9(input):
    ncc.clear()

    ncc.save_input_array('test', input)
    onnx_importer.utils.save(module, torch.from_numpy(input), opset_version=9)
    ncc.save_expect_array('test', onnx_importer.utils.run(input))

    onnx_importer.utils.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.05)

Esempio n. 6

def init_values():
    input = np.asarray([1, -2, 3, 4, -9, 0],
                       dtype=np.float32).reshape([1, 2, -1])
    ncc.save_input_array('test', input)
    ncc.save_expect_array('test', ncc.run_tflite(input))

Esempio n. 7

File: test_simple.py Progetto: zhen8838/nncase

def init_values():
    input = np.asarray([1.], dtype=np.float32)
    ncc.save_input_array('test', input)
    ncc.save_expect_array('test', ncc.run_tflite(input[0]))