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