Searched refs:grid_len (Results 1 – 3 of 3) sorted by relevance
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_android_apk_demo/app/src/main/cpp/ |
| H A D | post_process.cc | 149 int grid_len = grid_h * grid_w; in process() local
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()
164 … float box_y = sigmoid(deqnt_affine_to_f32(in_ptr[grid_len], zp, scale)) * 2.0 - 0.5; in process()
165 … float box_w = sigmoid(deqnt_affine_to_f32(in_ptr[2 * grid_len], zp, scale)) * 2.0; in process()
166 … float box_h = sigmoid(deqnt_affine_to_f32(in_ptr[3 * grid_len], zp, scale)) * 2.0; in process()
174 int8_t maxClassProbs = in_ptr[5 * grid_len]; in process()
178 int8_t prob = in_ptr[(5 + k) * grid_len]; in process()
|
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_demo/src/ |
| H A D | postprocess.cc | 194 int grid_len = grid_h * grid_w; in process() local
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()
205 float box_y = sigmoid(deqnt_affine_to_f32(in_ptr[grid_len], zp, scale)) * 2.0 - 0.5; in process()
206 float box_w = sigmoid(deqnt_affine_to_f32(in_ptr[2 * grid_len], zp, scale)) * 2.0; in process()
207 float box_h = sigmoid(deqnt_affine_to_f32(in_ptr[3 * grid_len], zp, scale)) * 2.0; in process()
215 int8_t maxClassProbs = in_ptr[5 * grid_len]; in process()
218 int8_t prob = in_ptr[(5 + k) * grid_len]; in process()
|
| /OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_yolov5_demo/src/ |
| H A D | postprocess.cc | 213 int grid_len = grid_h * grid_w; in process() local
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()
228 … float box_y = sigmoid(deqnt_affine_to_f32(in_ptr[grid_len], zp, scale)) * 2.0 - 0.5; in process()
229 … float box_w = sigmoid(deqnt_affine_to_f32(in_ptr[2 * grid_len], zp, scale)) * 2.0; in process()
230 … float box_h = sigmoid(deqnt_affine_to_f32(in_ptr[3 * grid_len], zp, scale)) * 2.0; in process()
238 int8_t maxClassProbs = in_ptr[5 * grid_len]; in process()
242 int8_t prob = in_ptr[(5 + k) * grid_len]; in process()
|