|
|
/*********版权所有（C）2024，武汉高德红外股份有限公司***************
|
|
|
* 文件名称： Arith_ObjArrest.cpp
|
|
|
* 文件标识： 跟踪目标重捕时机的返回
|
|
|
* 内容摘要：
|
|
|
* 其它说明： "Arith_ObjArrest.cpp"的函数、全局变量、宏定义，统一前缀为简写"ObjArrest_"
|
|
|
* 当前版本： 固化V2.0
|
|
|
* 创建作者： lk
|
|
|
* 创建日期： 2024.01.23
|
|
|
*************************************************************************/
|
|
|
#ifndef __ARITH_OBJ_ARREST_H__
|
|
|
#define __ARITH_OBJ_ARREST_H__
|
|
|
|
|
|
#include "Arith_Common.hpp"
|
|
|
#include "KCF/API_KCF_Tracker.h"
|
|
|
#include "Arith_AITTracker.h"
|
|
|
#include "Arith_OccludeJudge.h"
|
|
|
|
|
|
class TargetArrest
|
|
|
{
|
|
|
public:
|
|
|
TargetArrest(OccJudge* pOccFea, API_KCF_Tracker* pKCFTracker, API_TLD* pTLDTracker);
|
|
|
~TargetArrest();
|
|
|
|
|
|
public:
|
|
|
// 与遮挡重捕关联的遮挡判断事件
|
|
|
OccJudge* m_pOccFea;
|
|
|
|
|
|
API_KCF_Tracker* m_pKCFTracker;
|
|
|
API_TLD* m_pTLDTracker;
|
|
|
|
|
|
//重捕调试参数
|
|
|
UBYTE8 m_nArrestKCFStatus;
|
|
|
UBYTE8 m_nArrestAIStatus;
|
|
|
FLOAT32 m_fArrestKcfRes;
|
|
|
|
|
|
//重捕参数设置
|
|
|
BBOOL m_bEnableArrestCorr;
|
|
|
BBOOL m_bEnableArrestAngle;
|
|
|
BBOOL m_bEnableAreestEsay;
|
|
|
SINT32 m_nArrestEsayCnt;
|
|
|
|
|
|
//KCF重捕模块
|
|
|
void TLD_RecaptureKCF_Run(GD_VIDEO_FRAME_S img, PIPE* pLockingPipe,GLB_INPUT* p_GLB_Input);
|
|
|
|
|
|
//AI重捕模块
|
|
|
void TLD_RecaptureDasiamRPN_Run(GD_VIDEO_FRAME_S img, PIPE* pLockingPipe, GLB_INPUT* p_GLB_Input, AIT_Interface* pAITracker);
|
|
|
|
|
|
void AID_RecaptureKCF_Run(GD_VIDEO_FRAME_S img, PIPE* pLockingPipe, GLB_INPUT* p_GLB_Input);
|
|
|
|
|
|
//重捕成功参数更新
|
|
|
void ObjArrest_CleanUpObjArrestInfo(void);
|
|
|
|
|
|
//初始化解锁清理
|
|
|
void ObjArrest_CleanUpAllKCFArrestInfo(void);
|
|
|
|
|
|
|
|
|
//设置重捕参数
|
|
|
void ObjArrest_setArrestCorrPara(BBOOL bEnableArrestCorr);
|
|
|
|
|
|
void ObjArrest_setArrestAnglePara(BBOOL bEnableArrestAngle);
|
|
|
|
|
|
void ObjArrest_setArrestEsayPara(BBOOL bEnableArrestEsay);
|
|
|
|
|
|
void ObjArrest_setArrestEsayCntPara(SINT32 nArrestEsayCnt);
|
|
|
|
|
|
|
|
|
private:
|
|
|
UINT32 g_GLB_nMaxConfContinueFrame; //最大NN置信度处，相关性确认的连续帧数，避免出现错误重捕
|
|
|
UINT32 g_GLB_nConfShortConfirmContinueFrame; // 相关性确认的连续帧数，避免出现错误重捕
|
|
|
UINT32 g_GLB_nConfLongConfirmContinueFrame; // 相关性确认的连续帧数，避免出现错误重捕
|
|
|
|
|
|
MINMAXRECT brCurrentBoxTLDPre = { 0 }; //记录上一帧TLD检测位置
|
|
|
MINMAXRECT brCurrentBoxNNPre = { 0 };
|
|
|
MINMAXRECT brCurrentBoxNNPreAI = { 0 };
|
|
|
|
|
|
|
|
|
UBYTE8 m_bRecaptureKCF; //TLD_KCF重捕成功标志位
|
|
|
UBYTE8 m_bRecaptureAI; //TLD_DaSiamRPN重捕成功标志位
|
|
|
UINT32 m_ArrestCntKCF; //TLD_KCF重捕帧计数
|
|
|
UINT32 m_ArrestCntAI; //TLD_DaSiamRPN重捕帧计数
|
|
|
UBYTE8 m_nContiFrameCnt; //DaSiamRPN重捕连续帧计数
|
|
|
UBYTE8 m_nNNContiFrameCnt; //第二条成功重捕连续帧计数
|
|
|
SINT32 m_nDelayArrestCnt; //延时重捕帧计数
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
#endif |
