planning: add instance num stats.

fzf123 · Feb 6, 2020 · c518b37 · c518b37
1 parent 0485ffe
commit c518b37
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/modules/planning/pipeline/feature_generator.cc b/modules/planning/pipeline/feature_generator.cc
@@ -67,8 +67,10 @@ void FeatureGenerator::WriteOutInstances(const Instances& instances,
 void FeatureGenerator::Close() {
   const std::string file_name = absl::StrCat(
       FLAGS_planning_data_dir, "/instances.", instance_file_index_, ".bin");
+  total_instance_num_ += instances_.instances_size();
   WriteOutInstances(instances_, file_name);
   ++instance_file_index_;
+  AINFO << "Total instance number:" << total_instance_num_;
 }
 
 void FeatureGenerator::GenerateTrajectoryLabel(
@@ -126,6 +128,7 @@ void FeatureGenerator::OnLocalization(
     const std::string file_name = absl::StrCat(
         FLAGS_planning_data_dir, "/instances.", instance_file_index_, ".bin");
     WriteOutInstances(instances_, file_name);
+    total_instance_num_ += instances_.instances_size();
     instances_.Clear();
     ++instance_file_index_;
     instance_ = instances_.add_instances();

diff --git a/modules/planning/pipeline/feature_generator.h b/modules/planning/pipeline/feature_generator.h
@@ -46,7 +46,9 @@ class FeatureGenerator {
   Instance* instance_ = nullptr;  // not owned
   Instances instances_;
   int instance_file_index_ = 0;
-  std::list<apollo::localization::LocalizationEstimate> localization_for_label_;
+  std::list<apollo::localization::LocalizationEstimate>
+      localization_for_label_;
+  int total_instance_num_ = 0;
 };
 
 }  // namespace planning