mirror of
https://github.com/Ascend/ascend_community_projects.git
synced 2025-09-26 20:01:17 +08:00
删除文件 RosSdkInfer/temp.cpp
This commit is contained in:
@@ -1,50 +0,0 @@
|
||||
|
||||
APP_ERROR MxpiMotSimpleSortBase::GetModelInferResult(MxpiBuffer &buffer, std::vector<DetectObject> &detectObjectList)
|
||||
{
|
||||
MxpiMetadataManager mxpiMetadataManager(buffer);
|
||||
APP_ERROR ret = CheckDataStructure(mxpiMetadataManager);
|
||||
if (ret != APP_ERR_OK) {
|
||||
LogError << errorInfo_.str();
|
||||
return ret;
|
||||
}
|
||||
std::shared_ptr<MxpiObjectList> objectList = std::static_pointer_cast<MxpiObjectList>(
|
||||
mxpiMetadataManager.GetMetadata(dataSourceDetection_));
|
||||
std::shared_ptr<MxpiFeatureVectorList> featureList;
|
||||
if (objectList->objectvec_size() == 0) {
|
||||
LogDebug << "Object detection result of model infer is null.";
|
||||
return APP_ERR_OK;
|
||||
}
|
||||
if (withFeature_) {
|
||||
featureList = std::static_pointer_cast<MxpiFeatureVectorList>(
|
||||
mxpiMetadataManager.GetMetadata(dataSourceFeature_));
|
||||
if (featureList->featurevec_size() == 0) {
|
||||
ret = APP_ERR_COMM_FAILURE;
|
||||
errorInfo_ << GetError(ret, elementName_) << "Face short feature result of model infer is null.";
|
||||
return ret;
|
||||
}
|
||||
}
|
||||
for (int i = 0; i < objectList->objectvec_size(); ++i) {
|
||||
DetectObject detectObject {};
|
||||
detectObject.detectInfo = objectList->objectvec(i);
|
||||
detectObject.memberId = static_cast<uint32_t>(i);
|
||||
if (withFeature_) {
|
||||
GetFeatureVector(featureList, i, detectObject);
|
||||
}
|
||||
detectObjectList.push_back(detectObject);
|
||||
}
|
||||
return APP_ERR_OK;
|
||||
}
|
||||
|
||||
void MxpiMotSimpleSortBase::GetFeatureVector(const std::shared_ptr<MxTools::MxpiFeatureVectorList> &featureList,
|
||||
const int32_t &memberId, DetectObject &detectObject)
|
||||
{
|
||||
for (int i = 0; i < featureList->featurevec_size(); ++i) {
|
||||
if (featureList->featurevec(i).headervec_size() == 0) {
|
||||
LogError << GetError(APP_ERR_COMM_OUT_OF_RANGE) << "protobuf message vector is invalid.";
|
||||
return;
|
||||
}
|
||||
if (featureList->featurevec(i).headervec(0).memberid() == memberId) {
|
||||
detectObject.featureVector = featureList->featurevec(i);
|
||||
}
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user