diff --git a/inference/modelInference/sampleYOLOV7MultiInput/src/main.cpp b/inference/modelInference/sampleYOLOV7MultiInput/src/main.cpp index 7fd44e01a2cae2ff48663c968b5ec6233a64149f..ec53f76df9965317bb3c6809ef6115279b12435c 100644 --- a/inference/modelInference/sampleYOLOV7MultiInput/src/main.cpp +++ b/inference/modelInference/sampleYOLOV7MultiInput/src/main.cpp @@ -163,6 +163,7 @@ void CreateALLThreadInstance(vector& threadTbl, AclLiteResou detectPostParam.threadInstName.assign(postName.c_str()); detectPostParam.context = context; detectPostParam.runMode = runMode; + dataInputParam.queueSize = kMsgQueueSize; threadTbl.push_back(detectPostParam); } @@ -171,6 +172,7 @@ void CreateALLThreadInstance(vector& threadTbl, AclLiteResou dataOutputParam.threadInstName.assign(dataOutputName.c_str()); dataOutputParam.context = context; dataOutputParam.runMode = runMode; + dataInputParam.queueSize = kMsgQueueSize; threadTbl.push_back(dataOutputParam); if (outputType == "rtsp") @@ -180,6 +182,7 @@ void CreateALLThreadInstance(vector& threadTbl, AclLiteResou rtspDisplayThreadParam.threadInstName.assign(rtspDisplayName.c_str()); rtspDisplayThreadParam.context = context; rtspDisplayThreadParam.runMode = runMode; + dataInputParam.queueSize = kMsgQueueSize; threadTbl.push_back(rtspDisplayThreadParam); } kExitCount++;