@@ -309,7 +309,9 @@ namespace op
309309 // Note: We realized that somehow doing it on GPU for any number of GPUs does speedup the whole OP
310310 resizeOnCpu = false ;
311311 addCvMatToOpOutputInCpu = addCvMatToOpOutput
312- && (resizeOnCpu || !renderOutputGpu || wrapperStructPose.poseMode != PoseMode::Enabled);
312+ && (resizeOnCpu || !renderOutputGpu || wrapperStructPose.poseMode != PoseMode::Enabled
313+ // Resize in GPU causing bug
314+ || wrapperStructPose.outputSize .x != -1 || wrapperStructPose.outputSize .y != -1 );
313315 if (addCvMatToOpOutputInCpu)
314316 {
315317 const auto gpuResize = false ;
@@ -329,7 +331,8 @@ namespace op
329331 wrapperStructPose.poseModel , modelFolder, gpuId + gpuNumberStart,
330332 wrapperStructPose.heatMapTypes , wrapperStructPose.heatMapScaleMode ,
331333 wrapperStructPose.addPartCandidates , wrapperStructPose.maximizePositives ,
332- wrapperStructPose.protoTxtPath .getStdString (), wrapperStructPose.caffeModelPath .getStdString (),
334+ wrapperStructPose.protoTxtPath .getStdString (),
335+ wrapperStructPose.caffeModelPath .getStdString (),
333336 wrapperStructPose.upsamplingRatio , wrapperStructPose.poseMode == PoseMode::Enabled,
334337 wrapperStructPose.enableGoogleLogging
335338 ));
0 commit comments