Home
last modified time | relevance | path

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

/OK3568_Linux_fs/external/rknpu2/examples/rknn_yolov5_demo/src/
H A Dmain_video.cc46 int model_channel; member
210 app_ctx->model_channel = input_attrs[0].dims[1]; in init_model()
217 app_ctx->model_channel = input_attrs[0].dims[3]; in init_model()
219 …=%d, width=%d, channel=%d\n", app_ctx->model_height, app_ctx->model_width, app_ctx->model_channel); in init_model()
239 int model_channel = app_ctx->model_channel; in inference_model() local
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()