Home
last modified time | relevance | path

Searched refs:model_height (Results 1 – 2 of 2) sorted by relevance

/OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_demo/src/
H A Dmain_video.cc48 int model_height; member
211 app_ctx->model_height = input_attrs[0].dims[2]; in init_model()
215 app_ctx->model_height = input_attrs[0].dims[1]; in init_model()
219 …printf("model input height=%d, width=%d, channel=%d\n", app_ctx->model_height, app_ctx->model_widt… in init_model()
238 int model_height = app_ctx->model_height; in inference_model() local
259 float scale_h = (float)model_height / img->height; 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 Dmain.cc445 int model_height = 0; in main() local
450 model_height = input_attrs[0].dims[3]; in main()
456 model_height = input_attrs[0].dims[2]; in main()
460 float scale_h = (float)model_height / img_height; in main()