|
# Config for Model Input PreProcess
rknn.config(channel_mean_value='0 0 0 255', reorder_channel='0 1 2')
# Load TensorFlow Model
print('--> Loading model')
rknn.load_tensorflow(tf_pb='./mtcnn_1.12.pb',
inputs=['input'],
outputs=['prob', 'landmarks', 'box'],
input_size_list=[[INPUT_SIZE, INPUT_SIZE, 3]])
print('done')
请问inputs和outputs该怎么写???
|
|