Home
last modified time | relevance | path

Searched refs:grid_h (Results 1 – 5 of 5) sorted by relevance

/OK3568_Linux_fs/external/rknn-toolkit2/examples/darknet/yolov3_416x416/
H A Dyolov3_utils.py25 grid_h, grid_w = map(int, input.shape[0:2])
37 row = np.tile(np.arange(0, grid_h).reshape(-1, 1), grid_h)
39 col = col.reshape(grid_h, grid_w, 1, 1).repeat(3, axis=-2)
40 row = row.reshape(grid_h, grid_w, 1, 1).repeat(3, axis=-2)
44 box_xy /= (grid_w, grid_h)
/OK3568_Linux_fs/external/rknn-toolkit2/examples/onnx/yolov5/
H A Dtest.py47 grid_h, grid_w = map(int, input.shape[0:2])
57 row = np.tile(np.arange(0, grid_h).reshape(-1, 1), grid_h)
58 col = col.reshape(grid_h, grid_w, 1, 1).repeat(3, axis=-2)
59 row = row.reshape(grid_h, grid_w, 1, 1).repeat(3, axis=-2)
62 box_xy *= int(IMG_SIZE/grid_h)
/OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_android_apk_demo/app/src/main/cpp/
H A Dpost_process.cc143 static int process(int8_t *input, int *anchor, int grid_h, int grid_w, int height, int width, int s… in process() argument
149 int grid_len = grid_h * grid_w; in process()
154 for (int i = 0; i < grid_h; i++) in process()
/OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_demo/src/
H A Dpostprocess.cc189 static int process(int8_t* input, int* anchor, int grid_h, int grid_w, int height, int width, int s… in process() argument
194 int grid_len = grid_h * grid_w; in process()
198 for (int i = 0; i < grid_h; i++) { in process()
/OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_yolov5_demo/src/
H A Dpostprocess.cc207 static int process(int8_t *input, int *anchor, int grid_h, int grid_w, int height, int width, int s… in process() argument
213 int grid_len = grid_h * grid_w; in process()
218 for (int i = 0; i < grid_h; i++) in process()