SiamGATTracker()学习

本文详细介绍了SiamGATTracker类的实现,该类用于目标跟踪,通过模型预测目标在视频序列中的位置。核心函数包括init和track,前者用于初始化,后者处理后续帧的跟踪。跟踪过程中考虑了上下文区域、通道平均值、搜索窗口大小和惩罚策略,以适应目标尺度变化。同时,强调了在使用预训练模型进行测试时,需调用model.eval()以避免权重更新。

tracker()函数的作用就是传入video sequence和first frame中的ground truth bbox,然后通过模型,得到后续帧的目标位置,主要通过两个函数实现:init和tracker

  • initi:当idx == 0,即传入第一帧图片时,初始化参数,计算一些之后搜索区域的中心等
  • tracker:idx不等于0,传入后续的帧,根据网络返回目标的box坐标。
    class SiamGATTracker(SiameseTracker):
        def __init__(self, model):
            super(SiamGATTracker, self).__init__()
            self.score_size = cfg.TRACK.SCORE_SIZE #=====25
            hanning = np.hanning(self.score_size)
            self.window = np.outer(hanning, hanning)
            self.model = model
            self.model.eval()

训练完train_datasets之后,model要来测试样本了。在model(test_datasets)之前,需要加上model.eval(). 否则的话,有输入数据,即使不训练,它也会改变权值。这是model中含有batch normalization层所带来的的性质。 

    def init(self, img, bbox):
        
        self.center_pos = np.array([bbox[0]+(bbox[2]-1)/2,
                                    bbox[1]+(bbox[3]-1)/2])
     
评论 2
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值