Searched refs:input2_data (Results 1 – 2 of 2) sorted by relevance
| /OK3568_Linux_fs/external/rknn-toolkit2/examples/darknet/yolov3_416x416/ |
| H A D | test.py | 77 input2_data = outputs[2] variable
82 input2_data = input2_data.reshape(SPAN, LISTSIZE, GRID2, GRID2) variable
87 input_data.append(np.transpose(input2_data, (2, 3, 0, 1)))
|
| /OK3568_Linux_fs/external/rknn-toolkit2/examples/onnx/yolov5/ |
| H A D | test.py | 296 input2_data = outputs[2] variable
300 input2_data = input2_data.reshape([3, -1]+list(input2_data.shape[-2:])) variable
305 input_data.append(np.transpose(input2_data, (2, 3, 0, 1)))
|