cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix DetectionUniqueIdCalculator consume prob

Browse Source
This commit is contained in:
william.ywn 2020-11-05 11:06:59 +08:00
parent f15da632de
commit 7ee671aba1
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mediapipe/calculators/util/detection_unique_id_calculator.cc
View File

@ -33,8 +33,8 @@ inline int GetNextDetectionId() { return ++detection_id; }
} // namespace } // namespace
// Assign a unique id to detections. // Assign a unique id to detections.
// Note that the calculator will consume the input vector of Detection or // Note that the calculator will consume or copy the input vector of Detection or
// DetectionList. So the input stream can not be connected to other calculators. // DetectionList.
// //
// Example config: // Example config:
// node { // node {
@ -76,7 +76,7 @@ REGISTER_CALCULATOR(DetectionUniqueIdCalculator);
if (cc->Inputs().HasTag(kDetectionListTag) && if (cc->Inputs().HasTag(kDetectionListTag) &&
!cc->Inputs().Tag(kDetectionListTag).IsEmpty()) { !cc->Inputs().Tag(kDetectionListTag).IsEmpty()) {
auto result = auto result =
cc->Inputs().Tag(kDetectionListTag).Value().Consume<DetectionList>(); cc->Inputs().Tag(kDetectionListTag).Value().ConsumeOrCopy<DetectionList>();
if (result.ok()) { if (result.ok()) {
auto detection_list = std::move(result).ValueOrDie(); auto detection_list = std::move(result).ValueOrDie();
for (Detection& detection : *detection_list->mutable_detection()) { for (Detection& detection : *detection_list->mutable_detection()) {
@ -93,7 +93,7 @@ REGISTER_CALCULATOR(DetectionUniqueIdCalculator);
auto result = cc->Inputs() auto result = cc->Inputs()
.Tag(kDetectionsTag) .Tag(kDetectionsTag)
.Value() .Value()
.Consume<std::vector<Detection>>(); .ConsumeOrCopy<std::vector<Detection>>();
if (result.ok()) { if (result.ok()) {
auto detections = std::move(result).ValueOrDie(); auto detections = std::move(result).ValueOrDie();
for (Detection& detection : *detections) { for (Detection& detection : *detections) {