|
|
#pragma once
|
|
|
// 测试用跟踪器算法
|
|
|
#include "Arith_Common.hpp"
|
|
|
#include "Arith_TrackSAObj.h"
|
|
|
#include "KCF/API_KCF_Tracker.h"
|
|
|
#include "Arith_BkgMonitor.h"
|
|
|
#include "Arith_OccludeJudge.h"
|
|
|
#include "Arith_ObjArrest.h"
|
|
|
#include "Arith_AI_Tracker.h"
|
|
|
#include "TLD/API_TLD.h"
|
|
|
#include "Arith_Match.h"
|
|
|
#include "Arith_OccJudgeAI.h"
|
|
|
#include "Arith_AIDMonitor.h"
|
|
|
#include "PIPE/API_MOT_PipeProc.h"
|
|
|
#include "Arith_TrackNDYT.h"
|
|
|
|
|
|
|
|
|
struct Param_CustomTracker
|
|
|
{
|
|
|
/* data */
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
class CustomTracker
|
|
|
{
|
|
|
public:
|
|
|
CustomTracker(SINT32 nWidth, SINT32 nHeight);
|
|
|
~CustomTracker();
|
|
|
|
|
|
Param_CustomTracker mTrakingPara;
|
|
|
|
|
|
// 跟踪初始化：统一使用管道作为接口
|
|
|
bool Init(GD_VIDEO_FRAME_S img, PIPE* pLockPipe, GLB_INPUT* p_GLB_Input);
|
|
|
|
|
|
// 跟踪
|
|
|
bool Track(GD_VIDEO_FRAME_S img, GLB_INPUT* p_GLB_Input, API_MOT_PIPE* g_GLB_PipeProc);
|
|
|
|
|
|
// 跟踪器解锁
|
|
|
void Cancle();
|
|
|
|
|
|
|
|
|
// 跟踪结果，用于输出
|
|
|
OBJECTSTATUS ObjStatus;
|
|
|
|
|
|
//KCF跟踪结果
|
|
|
OBJECTSTATUS ObjStatusKCF;
|
|
|
|
|
|
//AI跟踪结果
|
|
|
OBJECTSTATUS ObjStatusAI;
|
|
|
|
|
|
PIPE* m_LockingPipe;//锁定管道指针，指向PIPE队列锁定目标，初始化赋值
|
|
|
|
|
|
// 遮挡判断模块
|
|
|
OccJudge* m_pOccJudgeKCF;
|
|
|
|
|
|
OccJudgeAI* m_pOccJudgeAI;
|
|
|
|
|
|
TrackEvent tkEventKCF; //KCF跟踪事件处置
|
|
|
|
|
|
AID_Monitor* Ai_DMonitor; //AI Detection 管道监控模块
|
|
|
|
|
|
|
|
|
TrackEvent tkEventAI; //AI跟踪事件处置
|
|
|
|
|
|
API_MOT_PIPE* m_pGLB_MOT_PIPE; // 多目标跟踪管道，背景目标监控信息
|
|
|
|
|
|
SINT32 m_nKCFResetCnt;
|
|
|
SINT32 m_nAIResetCnt;
|
|
|
SINT32 m_nAISeResetCnt;
|
|
|
|
|
|
public:
|
|
|
|
|
|
// 跟踪融合
|
|
|
bool TrackDecision(GD_VIDEO_FRAME_S img, GLB_INPUT* p_GLB_Input);
|
|
|
|
|
|
bool TO_TrackDecisionOfMiddleObj_KCF_AI(GD_VIDEO_FRAME_S img, GLB_INPUT* p_GLB_Input);
|
|
|
|
|
|
// 跟踪事件处理
|
|
|
void EventProcess(TrackEvent tkEvent, TrackEvent pEventAI, GD_VIDEO_FRAME_S img, PIPE* pLockingPipe,GLB_INPUT* p_GLB_Input);
|
|
|
|
|
|
// 重捕获重置操作
|
|
|
void RecapRestProcess(TrackEvent tkEvent, GD_VIDEO_FRAME_S img, PIPE* pLockingPipe, GLB_INPUT* p_GLB_Input);
|
|
|
//对地跟踪器状态更新
|
|
|
bool GroundTrackKCF(GD_VIDEO_FRAME_S img, GLB_INPUT* p_GLB_Input, API_KCF_Tracker* pKCFTracker,
|
|
|
OccJudge* pOccJudge, TrackEvent* tkEventStatus, TLD_Para* pTLDPara);
|
|
|
|
|
|
bool GroundTrackAI(GLB_INPUT* p_GLB_Input, AIT_Interface* pAITracker, OccJudgeAI* pOccJudge,
|
|
|
TrackEvent* tkEventAI, TLD_Para* pTLDPara);
|
|
|
|
|
|
//kcf跟踪器（通用）
|
|
|
API_KCF_Tracker* m_pKCFTracker;
|
|
|
NCCAccuracy* m_pNccCorrect;
|
|
|
|
|
|
// TLD
|
|
|
API_TLD* m_pTLDTracker;
|
|
|
|
|
|
// 遮挡重捕模块
|
|
|
TargetArrest* m_pTargetArrest;
|
|
|
|
|
|
// AI跟踪器信息接口
|
|
|
AIT_Interface* m_pAITracker;
|
|
|
|
|
|
NDYT* m_pNDYTTracker;
|
|
|
|
|
|
public:
|
|
|
CENTERRECT getKCFTracker_SrBox();
|
|
|
|
|
|
// 获取TLD搜索波门
|
|
|
CENTERRECT getTLD_SrBox();
|
|
|
|
|
|
}; |
