商汤开源最大目标跟踪库PySOT,代码已正式上线!

栏目: Python · 发布时间: 4年前

内容简介:前几天 CVer推送一篇:本以为是2019年5月底会上线代码,但于2019年5月13日凌晨,PySOT已正式上线代码!

前几天 CVer推送一篇: 重磅!商汤开源最大目标跟踪库PySOT:含SiamRPN++和SiamMask等算法 ,介绍了来自商汤科技的STVIR(SenseTime Video Intelligence Research team)开源的目标跟踪库:PySOT。当时PySOT并没有上传代码,所以之前的文章仅介绍了新特性,但得到大家广泛关注。

本以为是2019年5月底会上线代码,但于2019年5月13日凌晨,PySOT已正式上线代码!

https://github.com/STVIR/pysot

商汤开源最大目标跟踪库PySOT,代码已正式上线!

之前Amusi也分享了几篇很棒关于目标跟踪很棒的文章,但本文要介绍的项目是paper+code,这个是大家喜闻乐见的。

PySOT特性

PySOT 目标跟踪库实现了目前SOTA的多个单目标跟踪算法,该项目是用 Python 编写,基于PyTorch深度学习框架。

该项目还包含用于评估跟踪器(tracker)的Python工具接口。

为什么叫 PySOT?大家应该能猜到是:Python/PyTorch + Single Object Tracking 的组合。Amusi 强烈建议后面再开源个 PyMOT,哈哈!

商汤开源最大目标跟踪库PySOT,代码已正式上线!

PySOT 共包含 5 种算法:

  • SiamMask (CVPR 2019)

  • SiamRPN++(CVPR 2019 oral)

  • DaSiamRPN (ECCV 2018)

  • SiamRPN(CVPR 2018)

  • SiamFC(ECCV 2016)

商汤开源最大目标跟踪库PySOT,代码已正式上线!

SiamRPN++

PySOT 提供 3 种 backbone:

  • ResNet{18, 34, 50}

  • MobileNetV2

  • AlexNet

PySOT 提供的评估 工具 支持如下数据集:

  • OTB2015

  • VOT16/18/19

  • VOT18-LT

  • LaSOT

  • UAV123

PySOT 安装与测试

最新更新的 PySOT 添加了安装教程、模型下载库、测试/评估脚本等内容。

PySOT 安装&环境依赖

  • Conda with Python 3.7.

  • Nvidia GPU.

  • PyTorch 0.4.1

  • yacs

  • pyyamlmatplotlib

  • tqdm

  • OpenCV

<code># Create environment and activate</code><code>conda create --name pysot python=3.7</code><code>conda activate pysot</code><code># Install numpy/pytorch/opencv</code><code>conda install numpy</code><code>conda install pytorch=0.4.1 torchvision cuda90 -c pytorch</code><code>pip install opencv-python</code><code># Install other requirements</code><code>pip install pyyaml yacs tqdm colorama matplotlib cython tensorboardX</code><code># Build extensions</code><code>python setup.py build_ext --inplace</code><code># Try with scripts</code><code>bash install.sh /path/to/your/conda pysot</code>

PySOT Model Zoo

Visual Tracking Baselines(Short-term Tracking)

商汤开源最大目标跟踪库PySOT,代码已正式上线!

PySOT 工具集:评估

商汤开源最大目标跟踪库PySOT,代码已正式上线!

如:VOT2018-LT

商汤开源最大目标跟踪库PySOT,代码已正式上线!

商汤开源最大目标跟踪库PySOT,代码已正式上线!

测试脚本

<code>cd experiments/siamrpn_r50_l234_dwxcorr</code><code>python -u ../../tools/test.py   \</code><code>  --snapshot model.pth   \ # model path</code><code>  --dataset VOT2018   \ # dataset name</code><code>  --config config.yaml    # config file</code>

侃侃

PySOT的代码已经上线,快快测试一下吧~


以上就是本文的全部内容,希望本文的内容对大家的学习或者工作能带来一定的帮助,也希望大家多多支持 码农网

查看所有标签

猜你喜欢:

本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们

.NET框架程序设计

.NET框架程序设计

(美)Jeffrey Richter、(美)Francesco Balena / 李建忠 / 华中科技大学出版社 / 2004-1 / 54.00元

Microsoft.NET框架为简化开发与卫联网无缝连接的应用程序和组件提供了强大的技术支持,如ASP.NET Web窗体、XML Web服务以及Windows窗体。本书的目的在于展示.NET框架中公共语言运行库存的核心内容。全书由两位广受尊敬的开发者/作者完成,并假设读者理解面向对象程序设计的基本概念,如数据抽象、继承和多态。书中内容清楚地解释了CLR的扩展类型系统,CLR如何管理类型的行为,以......一起来看看 《.NET框架程序设计》 这本书的介绍吧!

RGB CMYK 转换工具
RGB CMYK 转换工具

RGB CMYK 互转工具

HEX CMYK 转换工具
HEX CMYK 转换工具

HEX CMYK 互转工具