Home
last modified time | relevance | path

Searched refs:grid_w (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])
36 col = np.tile(np.arange(0, grid_w), grid_w).reshape(-1, grid_w)
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])
56 col = np.tile(np.arange(0, grid_w), grid_w).reshape(-1, grid_w)
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)
/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()
156 for (int j = 0; j < grid_w; j++) in process()
158 int8_t box_confidence = input[(PROP_BOX_SIZE * a + 4) * grid_len + i * grid_w + j]; in process()
161 int offset = (PROP_BOX_SIZE * a) * grid_len + i * grid_w + j; 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()
199 for (int j = 0; j < grid_w; j++) { in process()
200 int8_t box_confidence = input[(PROP_BOX_SIZE * a + 4) * grid_len + i * grid_w + j]; in process()
202 int offset = (PROP_BOX_SIZE * a) * grid_len + i * grid_w + j; 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()
220 for (int j = 0; j < grid_w; j++) in process()
222 int8_t box_confidence = input[(PROP_BOX_SIZE * a + 4) * grid_len + i * grid_w + j]; in process()
225 int offset = (PROP_BOX_SIZE * a) * grid_len + i * grid_w + j; in process()