Hello,
On Fri, 2 Feb 2024, at 08:26, wenbin.chen-at-intel....@ffmpeg.org wrote:
> +static void infer_completion_callback(void *args) {
> + THRequestItem *request = (THRequestItem*)args;
> + LastLevelTaskItem *lltask = request->lltask;
> + TaskItem *task = lltask->task;
> + DNNData outputs = { 0 };
> + THInferRequest *infer_request = request->infer_request;
> + THModel *th_model = (THModel *)task->model;
> + torch::Tensor *output = infer_request->output;
> +
> + c10::IntArrayRef sizes = output->sizes();
> + assert(sizes.size == 5);
Why 5?
> + outputs.order = DCO_RGB;
> + outputs.layout = DL_NCHW;
> + outputs.dims[2] = sizes.at(3);
> + outputs.dims[3] = sizes.at(4);
> + outputs.dt = DNN_FLOAT;
> + outputs.dims[1] = 3;
Why 3?
--
Jean-Baptiste Kempf - President
+33 672 704 734
_______________________________________________
ffmpeg-devel mailing list
ffmpeg-devel@ffmpeg.org
https://ffmpeg.org/mailman/listinfo/ffmpeg-devel
To unsubscribe, visit link above, or email
ffmpeg-devel-requ...@ffmpeg.org with subject "unsubscribe".
