Commit 86946029 authored by oscar's avatar oscar

提交gpu参数

parent 4b5f6f63
......@@ -17,6 +17,11 @@ class BaseTracker
public:
BaseTracker() {}
void SetGPU(int gpu)
{
m_isGPU = gpu;
}
int Run(const std::vector<std::vector<float> >& detections,std::vector<uint64_t>& detectionsId,std::map<uint64_t,int>& updateId,std::vector<uint64_t>& lostId);
std::map<uint64_t, std::shared_ptr<T> >& GetStates();
......@@ -28,13 +33,10 @@ public:
std::map<uint64_t, std::shared_ptr<T> > m_tracker;
//CalculateIouCallback m_cbIou = nullptr;
//float m_iou_threshold = 0.3f;//匈牙利算法大于这个iou值为配置成功
//int m_kMaxCoastCycles = 2;//卡尔曼算法大于这个值就删除
uint64_t m_countId = 0;//生成物体id的累加值
//std::map<uint64_t, StateInfo> m_states;
int m_isGPU = 0;//默认不使用
};
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment