From 3b94390346fd88c4f4e43be3da46184939931986 Mon Sep 17 00:00:00 2001 From: Gines Hidalgo Date: Sun, 26 Apr 2020 23:19:05 -0400 Subject: [PATCH] Fixed flag output_resolution not working with gpu rendering --- include/openpose/wrapper/wrapperAuxiliary.hpp | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/include/openpose/wrapper/wrapperAuxiliary.hpp b/include/openpose/wrapper/wrapperAuxiliary.hpp index 89e94bbd..9119c9ea 100644 --- a/include/openpose/wrapper/wrapperAuxiliary.hpp +++ b/include/openpose/wrapper/wrapperAuxiliary.hpp @@ -309,7 +309,9 @@ namespace op // Note: We realized that somehow doing it on GPU for any number of GPUs does speedup the whole OP resizeOnCpu = false; addCvMatToOpOutputInCpu = addCvMatToOpOutput - && (resizeOnCpu || !renderOutputGpu || wrapperStructPose.poseMode != PoseMode::Enabled); + && (resizeOnCpu || !renderOutputGpu || wrapperStructPose.poseMode != PoseMode::Enabled + // Resize in GPU causing bug + || wrapperStructPose.outputSize.x != -1 || wrapperStructPose.outputSize.y != -1); if (addCvMatToOpOutputInCpu) { const auto gpuResize = false; @@ -329,7 +331,8 @@ namespace op wrapperStructPose.poseModel, modelFolder, gpuId + gpuNumberStart, wrapperStructPose.heatMapTypes, wrapperStructPose.heatMapScaleMode, wrapperStructPose.addPartCandidates, wrapperStructPose.maximizePositives, - wrapperStructPose.protoTxtPath.getStdString(), wrapperStructPose.caffeModelPath.getStdString(), + wrapperStructPose.protoTxtPath.getStdString(), + wrapperStructPose.caffeModelPath.getStdString(), wrapperStructPose.upsamplingRatio, wrapperStructPose.poseMode == PoseMode::Enabled, wrapperStructPose.enableGoogleLogging )); -- GitLab