Searched refs:input1_data (Results 1 – 2 of 2) sorted by relevance
| /OK3568_Linux_fs/external/rknn-toolkit2/examples/darknet/yolov3_416x416/ |
| H A D | test.py | 75 input1_data = outputs[1] variable
76 np.save('./darknet_yolov3_416x416_1.npy', input1_data)
78 np.save('./darknet_yolov3_416x416_2.npy', input1_data)
81 input1_data = input1_data.reshape(SPAN, LISTSIZE, GRID1, GRID1) variable
86 input_data.append(np.transpose(input1_data, (2, 3, 0, 1)))
|
| /OK3568_Linux_fs/external/rknn-toolkit2/examples/onnx/yolov5/ |
| H A D | test.py | 295 input1_data = outputs[1] variable
299 input1_data = input1_data.reshape([3, -1]+list(input1_data.shape[-2:])) variable
304 input_data.append(np.transpose(input1_data, (2, 3, 0, 1)))
|