Searched refs:model_width (Results 1 – 2 of 2) sorted by relevance
| /OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_demo/src/ |
| H A D | main_video.cc | 47 int model_width; member
212 app_ctx->model_width = input_attrs[0].dims[3]; in init_model()
216 app_ctx->model_width = input_attrs[0].dims[2]; in init_model()
219 …ut height=%d, width=%d, channel=%d\n", app_ctx->model_height, app_ctx->model_width, app_ctx->model… in init_model()
237 int model_width = app_ctx->model_width; in inference_model() local
258 float scale_w = (float)model_width / img->width; in inference_model()
265 inputs[0].size = model_width * model_height * model_channel; in inference_model()
270 resize_buf = malloc(model_width * model_height * model_channel); in inference_model()
271 memset(resize_buf, 0, model_width * model_height * model_channel); in inference_model()
274 dst = wrapbuffer_virtualaddr((void*)resize_buf, model_width, model_height, RK_FORMAT_RGB_888); in inference_model()
[all …]
|
| /OK3568_Linux_fs/external/rknpu2/examples/RV1106_RV1103/rknn_yolov5_demo/src/ |
| H A D | main.cc | 444 int model_width = 0; in main() local
449 model_width = input_attrs[0].dims[2]; in main()
455 model_width = input_attrs[0].dims[1]; in main()
459 float scale_w = (float)model_width / img_width; in main()
|