Ubuntu16.04搭建YOLOv5+DeepSORT及训练自己的数据集

具体参考:(如果侵权,速删)

  1. https://zhuanlan.zhihu.com/p/354945895
  2. https://blog.csdn.net/qq_44703886/article/details/121327643
    3.https://blog.csdn.net/weixin_50008473/article/details/122347582
    4.https://zhuanlan.zhihu.com/p/50387521
    5.https://blog.csdn.net/fengdu78/article/details/119431155
    6.https://blog.csdn.net/didiaopao/article/details/120274519
    github:1.https://github.com/mikel-brostrom/Yolov5_DeepSort_Pytorch
    2.https://github.com/layumi/Person_reID_baseline_pytorch

环境:

Nvidia RTX 3060
Ubuntu 16.04
CUDA 11.1
cuDNN 8.2.0
torch 1.10.1+cu111
torchvision 0.11.2+cu111

一、搭建YOLOv5-DeepSORT虚拟环境

1.使用conda创建 YOLOv5-DeepSORT 需要的环境
conda create -n yolov5_deepsort python=3.8  //yolov5_deepsort是虚拟环境的名字
source activate yolov5_deepsort
2.下载YOLOv5-DeepSORT工程

YOLOv5-DeepSORT版本需要和YOLOv5版本对应,对应关系参考另一篇文章:https://www.jianshu.com/p/bfec81af1cf9 (亲测YOLOv5 v5.0 + DeepSORT v2.0 可运行)
(1)YOLOv5-DeepSORT官网地址:https://github.com/mikel-brostrom/Yolov5_DeepSort_Pytorch

//DeepSORT 官网版本
git clone https://github.com/mikel-brostrom/Yolov5_DeepSort_Pytorch

(2)YOLOv5-DeepSORT版本有很多,也可以点击 master - tag可以下载不同的版本


yolov5-deepsort-2.png

使用zip解压命令:unzip Yolov5_DeepSort_Pytorch-2.0(以v2.0为例)

3.安装依赖
cd Yolov5_DeepSort_Pytorch-2.0

注意:requirements.txt依赖项的Python 3.8 或更高版本,包括torch>=1.7.0,torchvision>=0.8.1
(1)可以使用pip命令安装

pip install -r requirements.txt  

(2)pytorch及各种import可自行安装,注意pytorch和cuda版本需对应,官网地址:https://pytorch.org/get-started/previous-versions/ 对应关系如下:

55pytorchCUDA.png

本文选择使用官网命令进行安装,torch 1.10.1+cu111,torchvision 0.11.2+cu111

pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html
yolov5-15.png

yolov5-deepsort4.png

import:
需要下载的包
(1)出现问题:ModuleNotFoundError: No module named 'cv2'

pip install opencv-python

(2)出现问题:ModuleNotFoundError: No module named 'pandas'

pip install pandas

(3)出现问题:ModuleNotFoundError: No module named 'requests'

pip install requests

(4)出现问题:ModuleNotFoundError: No module named 'yaml'

 pip install pyyaml

(5)出现问题:ModuleNotFoundError: No module named 'tqdm'

pip install tqdm

(6)出现问题:ModuleNotFoundError: No module named 'matplotlib'

pip install matplotlib

(7)出现问题:ModuleNotFoundError: No module named 'seaborn'

pip install seaborn

二、训练自己的数据

1.需要准备的如下文件

(1)YOLOv5权重文件,放在~/code/Yolov5_DeepSort_Pytorch-3.0/yolov5/runs/train/exp4/weights下的best.pt
(2)重识别/分类模型,放在~/code/Yolov5_DeepSort_Pytorch-3.0/deep_sort_pytorch/deep_sort/deep/checkpoint下的ckpt.t7
(3)待追踪的视频,放在~/code/Yolov5_DeepSort_Pytorch-3.0下的800-1.avi

2.执行命令:
cd  ~/code/Yolov5_DeepSort_Pytorch-3.0
python track.py --yolo_weights yolov5/weights/best.pt --source 800-1.mp4 --deep_sort_weights deep_sort_pytorch/deep_sort/deep/checkpoint/ckpt.t7 --device 0   --save-vid --save-txt
yolov5-deepsort11.png

YOLOv5 v2.0版本出现问题除了下载一堆包以外,遇到两个问题
问题1:RuntimeError: a view of a leaf Variable that requires grad is being used in an in-place operation.
解决方法参考:https://blog.csdn.net/weixin_44612221/article/details/115359866

问题2:AttributeError: Can't get attribute 'SPPF' on <module 'models.common' from '/home/slave110/code/yolov5-5.0/models/common.py'>
解决方法参考:https://www.bilibili.com/read/cv14568709

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 203,098评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,213评论 2 380
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 149,960评论 0 336
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,519评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,512评论 5 364
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,533评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,914评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,574评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,804评论 1 296
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,563评论 2 319
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,644评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,350评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,933评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,908评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,146评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,847评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,361评论 2 342

推荐阅读更多精彩内容