Project import generated by Copybara.

GitOrigin-RevId: f9a66589eaf652bb93f8e37ed9e4da26e59ef214
2020-10-19 13:23:25 -04:00
parent cccf6244d3
commit c828392681
68 changed files with 890 additions and 487 deletions
@@ -317,6 +317,7 @@ cc_library(
        "//mediapipe/util/tracking:box_tracker",
        "//mediapipe/util/tracking:tracking_visualization_utilities",
        "@com_google_absl//absl/container:flat_hash_set",
+        "@com_google_absl//absl/container:node_hash_map",
        "@com_google_absl//absl/container:node_hash_set",
        "@com_google_absl//absl/strings",
    ],
@@ -19,6 +19,7 @@
 #include <unordered_set>

 #include "absl/container/flat_hash_set.h"
+#include "absl/container/node_hash_map.h"
 #include "absl/container/node_hash_set.h"
 #include "absl/strings/numbers.h"
 #include "mediapipe/calculators/video/box_tracker_calculator.pb.h"
@@ -207,7 +208,7 @@ class BoxTrackerCalculator : public CalculatorBase {
  // Boxes that are tracked in streaming mode.
  MotionBoxMap streaming_motion_boxes_;

-  std::unordered_map<int, std::pair<TimedBox, TimedBox>> last_tracked_boxes_;
+  absl::node_hash_map<int, std::pair<TimedBox, TimedBox>> last_tracked_boxes_;
  int frame_num_since_reset_ = 0;

  // Cache used during streaming mode for fast forward tracking.
@@ -19,8 +19,6 @@ licenses(["notice"])

 package(default_visibility = ["//mediapipe/calculators/video:__subpackages__"])

-exports_files(["LICENSE"])
-
 proto_library(
    name = "flow_quantizer_model_proto",
    srcs = ["flow_quantizer_model.proto"],