Skip to content

Commit e30350c

Browse files
authored
refactor(autoware_multi_object_tracker): merge normal_vehicle_tracker and big_vehicle_tracker (#9613)
* refactor: define object_model_ at initialization * refactor: combine normal and big vehicle tracker
1 parent 3cfb03e commit e30350c

File tree

9 files changed

+41
-479
lines changed

9 files changed

+41
-479
lines changed

perception/autoware_multi_object_tracker/CMakeLists.txt

+1-2
Original file line numberDiff line numberDiff line change
@@ -36,8 +36,7 @@ set(${PROJECT_NAME}_lib
3636
lib/tracker/motion_model/ctrv_motion_model.cpp
3737
lib/tracker/motion_model/cv_motion_model.cpp
3838
lib/tracker/model/tracker_base.cpp
39-
lib/tracker/model/big_vehicle_tracker.cpp
40-
lib/tracker/model/normal_vehicle_tracker.cpp
39+
lib/tracker/model/vehicle_tracker.cpp
4140
lib/tracker/model/multiple_vehicle_tracker.cpp
4241
lib/tracker/model/bicycle_tracker.cpp
4342
lib/tracker/model/pedestrian_tracker.cpp

perception/autoware_multi_object_tracker/include/autoware/multi_object_tracker/tracker/model/multiple_vehicle_tracker.hpp

+3-4
Original file line numberDiff line numberDiff line change
@@ -20,9 +20,8 @@
2020
#define AUTOWARE__MULTI_OBJECT_TRACKER__TRACKER__MODEL__MULTIPLE_VEHICLE_TRACKER_HPP_
2121

2222
#include "autoware/kalman_filter/kalman_filter.hpp"
23-
#include "autoware/multi_object_tracker/tracker/model/big_vehicle_tracker.hpp"
24-
#include "autoware/multi_object_tracker/tracker/model/normal_vehicle_tracker.hpp"
2523
#include "autoware/multi_object_tracker/tracker/model/tracker_base.hpp"
24+
#include "autoware/multi_object_tracker/tracker/model/vehicle_tracker.hpp"
2625

2726
#include <rclcpp/time.hpp>
2827

@@ -32,8 +31,8 @@ namespace autoware::multi_object_tracker
3231
class MultipleVehicleTracker : public Tracker
3332
{
3433
private:
35-
NormalVehicleTracker normal_vehicle_tracker_;
36-
BigVehicleTracker big_vehicle_tracker_;
34+
VehicleTracker normal_vehicle_tracker_;
35+
VehicleTracker big_vehicle_tracker_;
3736

3837
public:
3938
MultipleVehicleTracker(

perception/autoware_multi_object_tracker/include/autoware/multi_object_tracker/tracker/model/normal_vehicle_tracker.hpp

-78
This file was deleted.

perception/autoware_multi_object_tracker/include/autoware/multi_object_tracker/tracker/model/big_vehicle_tracker.hpp perception/autoware_multi_object_tracker/include/autoware/multi_object_tracker/tracker/model/vehicle_tracker.hpp

+9-9
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,8 @@
1616
// Author: v1.0 Yukihiro Saito
1717
//
1818

19-
#ifndef AUTOWARE__MULTI_OBJECT_TRACKER__TRACKER__MODEL__BIG_VEHICLE_TRACKER_HPP_
20-
#define AUTOWARE__MULTI_OBJECT_TRACKER__TRACKER__MODEL__BIG_VEHICLE_TRACKER_HPP_
19+
#ifndef AUTOWARE__MULTI_OBJECT_TRACKER__TRACKER__MODEL__VEHICLE_TRACKER_HPP_
20+
#define AUTOWARE__MULTI_OBJECT_TRACKER__TRACKER__MODEL__VEHICLE_TRACKER_HPP_
2121

2222
#include "autoware/kalman_filter/kalman_filter.hpp"
2323
#include "autoware/multi_object_tracker/object_model/object_model.hpp"
@@ -27,16 +27,15 @@
2727
namespace autoware::multi_object_tracker
2828
{
2929

30-
class BigVehicleTracker : public Tracker
30+
class VehicleTracker : public Tracker
3131
{
3232
private:
33-
autoware_perception_msgs::msg::DetectedObject object_;
33+
object_model::ObjectModel object_model_;
3434
rclcpp::Logger logger_;
3535

36-
object_model::ObjectModel object_model_ = object_model::big_vehicle;
37-
3836
double velocity_deviation_threshold_;
3937

38+
autoware_perception_msgs::msg::DetectedObject object_;
4039
double z_;
4140

4241
struct BoundingBox
@@ -52,8 +51,9 @@ class BigVehicleTracker : public Tracker
5251
using IDX = BicycleMotionModel::IDX;
5352

5453
public:
55-
BigVehicleTracker(
56-
const rclcpp::Time & time, const autoware_perception_msgs::msg::DetectedObject & object,
54+
VehicleTracker(
55+
const object_model::ObjectModel & object_model, const rclcpp::Time & time,
56+
const autoware_perception_msgs::msg::DetectedObject & object,
5757
const geometry_msgs::msg::Transform & self_transform, const size_t channel_size,
5858
const uint & channel_index);
5959

@@ -75,4 +75,4 @@ class BigVehicleTracker : public Tracker
7575

7676
} // namespace autoware::multi_object_tracker
7777

78-
#endif // AUTOWARE__MULTI_OBJECT_TRACKER__TRACKER__MODEL__BIG_VEHICLE_TRACKER_HPP_
78+
#endif // AUTOWARE__MULTI_OBJECT_TRACKER__TRACKER__MODEL__VEHICLE_TRACKER_HPP_

perception/autoware_multi_object_tracker/include/autoware/multi_object_tracker/tracker/tracker.hpp

+1-2
Original file line numberDiff line numberDiff line change
@@ -20,13 +20,12 @@
2020
#define AUTOWARE__MULTI_OBJECT_TRACKER__TRACKER__TRACKER_HPP_
2121

2222
#include "model/bicycle_tracker.hpp"
23-
#include "model/big_vehicle_tracker.hpp"
2423
#include "model/multiple_vehicle_tracker.hpp"
25-
#include "model/normal_vehicle_tracker.hpp"
2624
#include "model/pass_through_tracker.hpp"
2725
#include "model/pedestrian_and_bicycle_tracker.hpp"
2826
#include "model/pedestrian_tracker.hpp"
2927
#include "model/tracker_base.hpp"
3028
#include "model/unknown_tracker.hpp"
29+
#include "model/vehicle_tracker.hpp"
3130

3231
#endif // AUTOWARE__MULTI_OBJECT_TRACKER__TRACKER__TRACKER_HPP_

0 commit comments

Comments
 (0)