VOT测试自己的数据集

  1. 导入环境:

    # 导入docker镜像 vot_toolkit.tar 路径在/nas_dataset/docker_images/vot_toolkit.tar
    docker load --input vot_toolkit.tar
    # 根据docker镜像建立容器
     sudo docker run --gpus all --shm-size=60g -it -v /你的本地项目路径:/test  -w /test vot_toolkit
    
  2. 生成配置文件
    在该目录下生成三个文件

    1. trackers.ini:

      [TRACKALL]  # <tracker-name>
      label = TRACKALL
      protocol = traxpython
      
      command = tracker_vot
      
      # Specify a path to trax python wrapper if it is not visible (separate by ; if using multiple paths)
      paths = /test
      
      # Additional environment paths
      env_PATH = <additional-env-paths>;${
             PATH}
      

      tracker-name(中括号那个),label,根据自己跟踪器情况而定。

      paths:运行文件所在目录

      command:运行跟踪器的启动文件(需要在上面的paths下有command.py这个文件。在上面这个例子就是需要在 文件目录 下面有tracker_vot.py这个文件。

      然后参照官方调试用跟踪器的示例文件,修改自己的跟踪器的启动文件,添加trax等通信代码。

    2. tracker_vot.py:
      模仿官方的tracker的输入输出去写即可

      class NCCTracker(object):
      
          def __init__(self, image, region):
          	"""
          	image: np.array->[h,w,3]
          	region: bbox
          	"""
          	
          	# 帧的初始化在init中完成 
              self.window = max(region.width, region.height) * 2
      		
              left = max(region.x, 0)
              top = max(region.y, 0)
      
              right = min(region.x + region.width, image.shape[1] - 1)
              bottom = min(region.y + region.height, image.shape[0] - 1)
      
              self.template = image[int(top):int(bottom), int(left):int(right)]
              self.position = (region.x + region.width / 2, region.y + region.height / 2)
              self.size = (region.width, region.height)
      
          def track(self, image):
      		
              left = max(round(self.position[0] - float
  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值