zoukankan      html  css  js  c++  java
  • deepsort + yolov5 轨迹追踪 pytorch 环境配置

    deepsort + yolov5 轨迹追踪 pytorch 环境配置

    简介

    SORT (Simple Online Realtime Tracking) 是基于目标检测的追踪。与基于底层图象特征(如ORB特征)相比,目标检测的重复检测更稳定,并且算法结构更简单。deepsort 基于sort框架上增加了基于深度学习的图象特征提取用于目标匹配,可以有效应对在目标交叉时ID交换和目标消失后重识别的问题。deepsort因为检测速度和精度都较好,在工业界有很多应用。deepsort 有很多开源的实现,以下是基于pytorch的 yolo_v5+deepsort 环境配置。

    配置过程

    1.下载代码

    代码地址:https://github.com/mikel-brostrom/Yolov5_DeepSort_Pytorch
    安装readme的 deepsort 和 yolo_v5的代码。

    2.下载 pytorch whl文件。

    pytorch 要用1.7.0以上版本。换源的版本没有最新的,只能在pytorch 官网用whl 文件安装,
    下载地址: https://download.pytorch.org/whl/torch_stable.html
    找到对应的torch和torchvision版本,上梯子把文件下下来。torch 要和系统和python 版本对应,如cp36是对应python3.6。torchvision要和torch的版本对应,查不到对应版本的化,安装torchvision时会显示对应的torch版本。

    3.配置环境

    因为要用特定的特定的python版本,可以在conda里创建虚拟环境。python 版本要与下载的torch版本相同

    conda create -n torh170 python=3.6
    source activate torh170
    

    环境名为 torch170,把轮子文件whl文件放在当前目录下安装

    pip install  torch-1.7.0-cp36-none-macosx_10_9_x86_64.whl  
    pip install torchvision-0.8.0-cp36-cp36m-macosx_10_9_x86_64.whl  
    

    安装其他依赖,可以用豆瓣源加速

    pip install opencv-python -i https://pypi.douban.com/simple/
    

    类似的安装其他依赖库 tqdm, pyyaml, matplotlib, easydict, scipy

    4.下载 模型weight文件和视频文件

    按照readme描述下载模型weight文件和视频文件,运行:

    python3 track.py -—source demo.avi
    

    5.MOTA 追踪结果指标

    MOT(Multi Object Tracking)Challenge是个公开的多目标追踪benchmark。可以用上吗的真值数据对追踪结果进行评价。MOT16官方的代码需要用到matlab,有人写了纯python的实现: https://github.com/shenh10/mot_evaluation
    结果和真值要写成对应的格式:

    <frame>, <id>, <bb_left>, <bb_top>, <bb_width>, <bb_height>, <conf>, <x>, <y>, <z>
    

    把结果和真值放在对应的文件夹内运行:

    python evaluate_tracking.py --gt ./data/MOT16-11/gt.txt --track ./data/MOT16-11/res.txt
    
  • 相关阅读:
    Java 替换 PPT 中的图片
    Java 转换 PDF 版本
    Java 在Word文档中添加艺术字
    Java 删除 Excel 中的空白行和列
    Java 为 PPT 中的图形添加阴影效果
    nginx作用及其配置
    Dockerfile
    Zookeeper学习
    vagrant up启动报错Call to WHvSetupPartition failed
    Vagrant安装启动linux,VitrtualBox,Centos7
  • 原文地址:https://www.cnblogs.com/dalaska/p/14153509.html
Copyright © 2011-2022 走看看