Home
last modified time | relevance | path

Searched refs:input0_data (Results 1 – 2 of 2) sorted by relevance

/OK3568_Linux_fs/external/rknn-toolkit2/examples/darknet/yolov3_416x416/
H A Dtest.py73 input0_data = outputs[0] variable
74 np.save('./darknet_yolov3_416x416_0.npy', input0_data)
80 input0_data = input0_data.reshape(SPAN, LISTSIZE, GRID0, GRID0) variable
85 input_data.append(np.transpose(input0_data, (2, 3, 0, 1)))
/OK3568_Linux_fs/external/rknn-toolkit2/examples/onnx/yolov5/
H A Dtest.py294 input0_data = outputs[0] variable
298 input0_data = input0_data.reshape([3, -1]+list(input0_data.shape[-2:])) variable
303 input_data.append(np.transpose(input0_data, (2, 3, 0, 1)))