18 lines
771 B
C++
18 lines
771 B
C++
#ifndef ASSOCIATION_H
|
|
#define ASSOCIATION_H
|
|
|
|
#include <Eigen/Dense>
|
|
#include <algorithm>
|
|
#include <vector>
|
|
#include "OCSortlapjv.h"
|
|
#define m_pi 3.1415926
|
|
|
|
namespace ANSOCSort {
|
|
std::tuple<Eigen::MatrixXf, Eigen::MatrixXf> speed_direction_batch(const Eigen::MatrixXf& dets,
|
|
const Eigen::MatrixXf& tracks);
|
|
Eigen::MatrixXf iou_batch(const Eigen::MatrixXf& bboxes1, const Eigen::MatrixXf& bboxes2);
|
|
Eigen::MatrixXf giou_batch(const Eigen::MatrixXf& bboxes1, const Eigen::MatrixXf& bboxes2);
|
|
std::tuple<std::vector<Eigen::Matrix<int, 1, 2>>, std::vector<int>, std::vector<int>> associate(Eigen::MatrixXf detections, Eigen::MatrixXf trackers, float iou_threshold, Eigen::MatrixXf velocities, Eigen::MatrixXf previous_obs_, float vdc_weight);
|
|
}
|
|
|
|
#endif |