本文地址：[https://blog.csdn.net/weixin_44936889/article/details/112002152](https://blog.csdn.net/weixin_44936889/article/details/112002152)
# 注意：
## 本项目使用Yolov5 3.0版本，最新版本5.0请移步：
[https://github.com/Sharpiless/yolov5-deepsort](https://github.com/Sharpiless/yolov5-deepsort)
## 注：新版本添加了类别显示功能
# 项目简介：
使用YOLOv5+Deepsort实现车辆行人追踪和计数，代码封装成一个Detector类，更容易嵌入到自己的项目中。
代码地址（欢迎star）：
[https://github.com/Sharpiless/Yolov5-deepsort-inference](https://github.com/Sharpiless/Yolov5-deepsort-inference)
最终效果：
# YOLOv5检测器：
```python
class Detector(baseDet):
def __init__(self):
super(Detector, self).__init__()
self.init_model()
self.build_config()
def init_model(self):
self.weights = 'weights/yolov5m.pt'
self.device = '0' if torch.cuda.is_available() else 'cpu'
self.device = select_device(self.device)
model = attempt_load(self.weights, map_location=self.device)
model.to(self.device).eval()
model.half()
# torch.save(model, 'test.pt')
self.m = model
self.names = model.module.names if hasattr(
model, 'module') else model.names
def preprocess(self, img):
img0 = img.copy()
img = letterbox(img, new_shape=self.img_size)[0]
img = img[:, :, ::-1].transpose(2, 0, 1)
img = np.ascontiguousarray(img)
img = torch.from_numpy(img).to(self.device)
img = img.half() # 半精度
img /= 255.0 # 图像归一化
if img.ndimension() == 3:
img = img.unsqueeze(0)
return img0, img
def detect(self, im):
im0, img = self.preprocess(im)
pred = self.m(img, augment=False)[0]
pred = pred.float()
pred = non_max_suppression(pred, self.threshold, 0.4)
pred_boxes = []
for det in pred:
if det is not None and len(det):
det[:, :4] = scale_coords(
img.shape[2:], det[:, :4], im0.shape).round()
for *x, conf, cls_id in det:
lbl = self.names[int(cls_id)]
if not lbl in ['person', 'car', 'truck']:
continue
x1, y1 = int(x[0]), int(x[1])
x2, y2 = int(x[2]), int(x[3])
pred_boxes.append(
(x1, y1, x2, y2, lbl, conf))
return im, pred_boxes
```
调用 self.detect 方法返回图像和预测结果
# DeepSort追踪器：
```python
deepsort = DeepSort(cfg.DEEPSORT.REID_CKPT,
max_dist=cfg.DEEPSORT.MAX_DIST, min_confidence=cfg.DEEPSORT.MIN_CONFIDENCE,
nms_max_overlap=cfg.DEEPSORT.NMS_MAX_OVERLAP, max_iou_distance=cfg.DEEPSORT.MAX_IOU_DISTANCE,
max_age=cfg.DEEPSORT.MAX_AGE, n_init=cfg.DEEPSORT.N_INIT, nn_budget=cfg.DEEPSORT.NN_BUDGET,
use_cuda=True)
```
调用 self.update 方法更新追踪结果
# 运行demo：
```bash
python demo.py
```
# 训练自己的模型：
参考我的另一篇博客：
[【小白CV】手把手教你用YOLOv5训练自己的数据集（从Windows环境配置到模型部署）](https://blog.csdn.net/weixin_44936889/article/details/110661862)
训练好后放到 weights 文件夹下
# 调用接口：
## 创建检测器：
```python
from AIDetector_pytorch import Detector
det = Detector()
```
## 调用检测接口：
```python
func_status = {}
func_status['headpose'] = None
result = det.feedCap(im, func_status)
```
其中 im 为 BGR 图像
返回的 result 是字典，result['frame'] 返回可视化后的图像
# 关注我的公众号：
感兴趣的同学关注我的公众号——可达鸭的深度学习教程：
# 联系作者：
> B站：[https://space.bilibili.com/470550823](https://space.bilibili.com/470550823)
> CSDN：[https://blog.csdn.net/weixin_44936889](https://blog.csdn.net/weixin_44936889)
> AI Studio：[https://aistudio.baidu.com/aistudio/personalcenter/thirdview/67156](https://aistudio.baidu.com/aistudio/personalcenter/thirdview/67156)
> Github：[https://github.com/Sharpiless](https://github.com/Sharpiless)
遵循 GNU General Public License v3.0 协议，标明目标检测部分来源：https://github.com/ultralytics/yolov5/
使用YOLOv5+Deepsort实现车辆行人追踪和计数（毕业设计） （103个子文件） 
.gitkeep 0B
train.jpg 59KB LICENSE 34KB README.md 5KB README.md 65B yolov5m.pt 41.92MB general.py 19KB json_logger.py 11KB yolo.py 10KB torch_utils.py 9KB metrics.py 8KB linear_assignment.py 8KB kalman_filter.py 8KB autoanchor.py 7KB train.py 6KB experimental.py 6KB nn_matching.py 5KB tracker.py 5KB google_utils.py 5KB track.py 5KB io.py 4KB common.py 4KB deep_sort.py 4KB evaluation.py 3KB original_model.py 3KB model.py 3KB iou_matching.py 3KB tracker.py 3KB test.py 2KB activations.py 2KB AIDetector_pytorch.py 2KB preprocessing.py 2KB feature_extractor.py 2KB detection.py 1KB demo.py 1KB draw.py 1KB BaseDetector.py 1KB parser.py 976B tools.py 734B __init__.py 500B log.py 463B asserts.py 316B evaluate.py 293B __init__.py 0B __init__.py 0B __init__.py 0B __init__.py 0B __init__.py 0B datasets.cpython-37.pyc 28KB plots.cpython-37.pyc 14KB general.cpython-36.pyc 14KB general.cpython-37.pyc 14KB torch_utils.cpython-37.pyc 9KB torch_utils.cpython-36.pyc 9KB yolo.cpython-37.pyc 9KB linear_assignment.cpython-37.pyc 7KB linear_assignment.cpython-36.pyc 7KB experimental.cpython-37.pyc 7KB metrics.cpython-37.pyc 7KB kalman_filter.cpython-36.pyc 7KB kalman_filter.cpython-37.pyc 7KB metrics.cpython-36.pyc 7KB nn_matching.cpython-36.pyc 6KB nn_matching.cpython-37.pyc 6KB autoanchor.cpython-37.pyc 6KB common.cpython-37.pyc 6KB loss.cpython-37.pyc 6KB track.cpython-37.pyc 5KB track.cpython-36.pyc 5KB tracker.cpython-37.pyc 5KB tracker.cpython-36.pyc 5KB deep_sort.cpython-37.pyc 4KB deep_sort.cpython-36.pyc 4KB iou_matching.cpython-37.pyc 3KB iou_matching.cpython-36.pyc 3KB google_utils.cpython-37.pyc 3KB model.cpython-37.pyc 3KB model.cpython-36.pyc 3KB google_utils.cpython-36.pyc 3KB feature_extractor.cpython-36.pyc 3KB feature_extractor.cpython-37.pyc 3KB preprocessing.cpython-37.pyc 2KB preprocessing.cpython-36.pyc 2KB detection.cpython-37.pyc 2KB detection.cpython-36.pyc 2KB BaseDetector.cpython-37.pyc 2KB parser.cpython-37.pyc 1KB parser.cpython-36.pyc 1KB __init__.cpython-37.pyc 676B __init__.cpython-36.pyc 672B __init__.cpython-37.pyc 231B __init__.cpython-37.pyc 231B __init__.cpython-36.pyc 227B __init__.cpython-36.pyc 227B __init__.cpython-37.pyc 222B __init__.cpython-37.pyc 221B __init__.cpython-37.pyc 220B __init__.cpython-36.pyc 218B __init__.cpython-36.pyc 127B ckpt.t7 43.9MB共 103 条
- 1
- 2
资源评论
Luminous5442024-07-22资源不错，很实用，内容全面，介绍详细，很好用，谢谢分享。
程序员无锋
- 粉丝: 3698
- 资源: 2563
下载权益
C知道特权
VIP文章
课程特权
VIP享7折，此内容立减5.97元 开通VIP
最新资源
- 基于Java和Vue的kopsoftKANBAN车间电子看板设计源码
- 影驰战将PS3111 东芝芯片TT18G23AIN开卡成功分享，图片里面画线的选项很重要
- 【C++初级程序设计·配套源码】第1期-语法基础
- 基于JavaScript、CSS、HTML的简易DOM版飞机游戏设计源码
- 基于Java开发的日程管理FlexTime应用设计源码
- SM2258XT-BGA144-4BGA180-6L-R1019 三星KLUCG4J1CB B0B1颗粒开盘工具 , EC, 3A, 94, 43, A4, CA 七彩虹SL300这个固件有用
- GJB 5236-2004 军用软件质量度量
- 30天开发操作系统 第 8 天 - 鼠标控制与切换32模式
- spice vd interface接口
- 安装Git时遇到找不到`/dev/null`的问题
资源上传下载、课程学习等过程中有任何疑问或建议，欢迎提出宝贵意见哦~我们会及时处理！
点击此处反馈
