Skip to content

Wuyang-Zhang/yolox-deepsort-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

58 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

yolox-deepsort-pytorch

nets、utils、model_data都是目标检测器所需要的文件 deepsort和shell文件是属于追踪的。其余文件夹可直接用相关路径代替。

shells文件夹中的detector.py问价根据检测器修改 只要输出坐标,类别,置信度与图片即可。

注意!!!

使用本文件时要根据自己的数据集的种类类别,需要修改model_data的voc_class.txt和coco_class数据集文件。

该文件所对应的目标检测器的网址为https://github.com/bubbliiiing/yolox-pytorch

bilibili:

https://www.bilibili.com/video/BV16S4y1b7MK/?spm_id_from=333.337.search-card.all.click&vd_source=e1de9f6d02128b9c85f5fdd03c7e72fc

github

根据https://github.com/renyuehe/bilibili-yolov5_deepsort 修改而来

tree 目录

image image

''' ├─deep_sort │├─configs ││ └─deepsort.yaml #deepsort配置文 │├─deep_sort ││ ├─deep ││ │ ├─checkpoint #deepsort的ReID网络权重 ││ │ ├─evaluate.py #ReID网络评估脚本 ││ │ ├─feature_extractor.py #特征提取器代码 ││ │ ├─model.py #ReID网络定义代码 ││ │ ├─prepare_car.py #整理车辆数据集脚本 ││ │ ├─prepare_person.py #整理行人数据集脚本 ││ │ ├─test.py #ReID网络测试脚本 ││ │ ├─train.py #ReID网络训练脚本 ││ ├─sort ││ │ ├─detection.py #Detection类定义代码 ││ │ ├─iou_matching.py #IoU计算代码 ││ │ ├─kalman_filter.py #卡尔曼滤波器代码 ││ │ ├─linear_assignment.py #分配算法代码 ││ │ ├─nn_matching.py #最近邻距离计算代码 ││ │ ├─preprocessing.py #非极大抑制代码 ││ │ ├─track.py #Track类定义代码,存储轨迹星系,其中包含轨迹狂位置和速度, ││ │ └─tracker.py #Tracker类定义代码 ││ └─deepsort.py #DeepSORT类定义代码 │└─utils #deepsort辅助工具代码 ├─models #yolov5 v5.0的models代码 ├─utils #yolov5 v5.0的utils代码 ├─shells │ ├─tools #工具类 │ ├─deepsortor #deepsort类封装 │ └─detecotr #yolov5类封装 ├─video #测试的视频文件 ├─weights #yolov5预训练权重 ├─demo_count_car.py #车辆计数演示脚本 ├─demo_count_person.py #行人计数演示脚本 └─demo_base.py #多目标跟踪演示脚本 '''

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages