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".