Project import generated by Copybara.
GitOrigin-RevId: b2062656e5b3d33264e28ed0cbca31c4b93fe1bf
This commit is contained in:
@@ -316,6 +316,7 @@ cc_library(
|
||||
"//mediapipe/util/tracking",
|
||||
"//mediapipe/util/tracking:box_tracker",
|
||||
"//mediapipe/util/tracking:tracking_visualization_utilities",
|
||||
"@com_google_absl//absl/container:node_hash_set",
|
||||
"@com_google_absl//absl/strings",
|
||||
],
|
||||
alwayslink = 1,
|
||||
|
||||
@@ -18,6 +18,7 @@
|
||||
#include <unordered_map>
|
||||
#include <unordered_set>
|
||||
|
||||
#include "absl/container/node_hash_set.h"
|
||||
#include "absl/strings/numbers.h"
|
||||
#include "mediapipe/calculators/video/box_tracker_calculator.pb.h"
|
||||
#include "mediapipe/framework/calculator_framework.h"
|
||||
@@ -193,12 +194,12 @@ class BoxTrackerCalculator : public CalculatorBase {
|
||||
TimedBoxProtoList initial_pos_;
|
||||
|
||||
// Keeps tracks boxes that have already been initialized.
|
||||
std::unordered_set<int> initialized_ids_;
|
||||
absl::node_hash_set<int> initialized_ids_;
|
||||
|
||||
// Non empty for batch mode tracking.
|
||||
std::string cache_dir_;
|
||||
// Ids to be tracked in batch_mode.
|
||||
std::unordered_set<int> batch_track_ids_;
|
||||
absl::node_hash_set<int> batch_track_ids_;
|
||||
|
||||
int frame_num_ = 0;
|
||||
|
||||
|
||||
Reference in New Issue
Block a user