Yolov5 deepsort inference，使用YOLOv5+Deepsort实现车辆行人追踪和计数，代码封装成一个Detector类，更容易嵌入到自己的项目中

Last update: Dec 31, 2022

Related tags

Deep Learning Yolov5-deepsort-inference

Overview

本文地址：https://blog.csdn.net/weixin_44936889/article/details/112002152

项目简介：

使用YOLOv5+Deepsort实现车辆行人追踪和计数，代码封装成一个Detector类，更容易嵌入到自己的项目中。

代码地址（欢迎star）：

https://github.com/Sharpiless/Yolov5-deepsort-inference

最终效果：

YOLOv5检测器：

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追踪器：

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：

python demo.py

训练自己的模型：

参考我的另一篇博客：

【小白CV】手把手教你用YOLOv5训练自己的数据集（从Windows环境配置到模型部署）

训练好后放到 weights 文件夹下

调用接口：

创建检测器：

from AIDetector_pytorch import Detector

det = Detector()

调用检测接口：

func_status = {}
func_status['headpose'] = None

result = det.feedCap(im, func_status)

其中 im 为 BGR 图像

返回的 result 是字典，result['frame'] 返回可视化后的图像

关注我的公众号：

感兴趣的同学关注我的公众号——可达鸭的深度学习教程：

联系作者：

遵循 GNU General Public License v3.0 协议，标明目标检测部分来源：https://github.com/ultralytics/yolov5/

You might also like...

A Python training and inference implementation of Yolov5 helmet detection in Jetson Xavier nx and Jetson nano

yolov5-helmet-detection-python A Python implementation of Yolov5 to detect head or helmet in the wild in Jetson Xavier nx and Jetson nano. In Jetson X

12 Dec 5, 2022

Rewrite ultralytics/yolov5 v6.0 opencv inference code based on numpy, no need to rely on pytorch

Rewrite ultralytics/yolov5 v6.0 opencv inference code based on numpy, no need to rely on pytorch; pre-processing and post-processing using numpy instead of pytroch.

21 Dec 12, 2022

Monocular 3D pose estimation. OpenVINO. CPU inference or iGPU (OpenCL) inference.

human-pose-estimation-3d-python-cpp RealSenseD435 (RGB) 480x640 + CPU Corei9 45 FPS (Depth is not used) 1. Run 1-1. RealSenseD435 (RGB) 480x640 + CPU

8 Oct 3, 2022

PyTorch-LIT is the Lite Inference Toolkit (LIT) for PyTorch which focuses on easy and fast inference of large models on end-devices.

PyTorch-LIT PyTorch-LIT is the Lite Inference Toolkit (LIT) for PyTorch which focuses on easy and fast inference of large models on end-devices. With

157 Dec 11, 2022

Data-depth-inference - Data depth inference with python

Welcome! This readme will guide you through the use of the code in this reposito

3 Feb 8, 2022

YOLOv5 in PyTorch ONNX CoreML TFLite

This repository represents Ultralytics open-source research into future object detection methods, and incorporates lessons learned and best practices evolved over thousands of hours of training and evolution on anonymized client datasets. All code and models are under active development, and are subject to modification or deletion without notice.

34.1k Dec 31, 2022

Comments

在运行初期已经出现了fps是10了，目标数是0的话，那我加的判断条件应该是什么呢？

在运行初期已经出现了fps是10了，目标数是0的话，那我加的判断条件应该是什么呢？ if COUNT == 0: break 这样吗，如果能解答就太感谢了，耽误您时间了，抱歉

Originally posted by @zzc-ai in https://github.com/Sharpiless/Yolov5-deepsort-inference/issues/1#issuecomment-755820307

您解决这个问题了吗？我遇到了相同的问题，如能解答，万分感谢！

opened by Iofal 1
为什么会有这个问题？

Traceback (most recent call last): File "E:/TuXiangChuLi/week14/Yolov5-deepsort-inference/demo.py", line 1, in from AIDetector_pytorch import Detector File "E:\TuXiangChuLi\week14\Yolov5-deepsort-inference\AIDetector_pytorch.py", line 6, in from utils.BaseDetector import baseDet File "E:\TuXiangChuLi\week14\Yolov5-deepsort-inference\utils\BaseDetector.py", line 1, in from tracker import update_tracker File "E:\TuXiangChuLi\week14\Yolov5-deepsort-inference\tracker.py", line 8, in cfg.merge_from_file("E:/TuXiangChuLi/week14/Yolov5-deepsort-inference/deep_sort/configs/deep_sort.yaml") File "E:\TuXiangChuLi\week14\Yolov5-deepsort-inference\deep_sort\utils\parser.py", line 23, in merge_from_file self.update(yaml.load(fo.read())) TypeError: load() missing 1 required positional argument: 'Loader'

Process finished with exit code 1

opened by HLKunn 1
how to train the deep sort model?

hello! thanks for your open source. but i want to know that how to train the deep sort model which can used in your repo? Could you give a instruction to us? THX

opened by brucelee78 0

Yolov5 deepsort inference，使用YOLOv5+Deepsort实现车辆行人追踪和计数，代码封装成一个Detector类，更容易嵌入到自己的项目中

Related tags

Overview

项目简介：

YOLOv5检测器：

DeepSort追踪器：

运行demo：

训练自己的模型：

调用接口：

创建检测器：

调用检测接口：

关注我的公众号：

联系作者：

You might also like...

A Python training and inference implementation of Yolov5 helmet detection in Jetson Xavier nx and Jetson nano

Rewrite ultralytics/yolov5 v6.0 opencv inference code based on numpy, no need to rely on pytorch

Monocular 3D pose estimation. OpenVINO. CPU inference or iGPU (OpenCL) inference.

PyTorch-LIT is the Lite Inference Toolkit (LIT) for PyTorch which focuses on easy and fast inference of large models on end-devices.

Data-depth-inference - Data depth inference with python

YOLOv5 in PyTorch ONNX CoreML TFLite

Drone detection using YOLOv5

joint detection and semantic segmentation, based on ultralytics/yolov5,

This repository is based on Ultralytics/yolov5, with adjustments to enable polygon prediction boxes.

Comments

在运行初期已经出现了fps是10了，目标数是0的话，那我加的判断条件应该是什么呢？

为什么会有这个问题？

how to train the deep sort model?

Owner

Torchserve server using a YoloV5 model running on docker with GPU and static batch inference to perform production ready inference.

🍅🍅🍅YOLOv5-Lite: lighter, faster and easier to deploy. Evolved from yolov5 and the size of model is only 1.7M (int8) and 3.3M (fp16). It can reach 10+ FPS on the Raspberry Pi 4B when the input size is 320×320~

TPH-YOLOv5: Improved YOLOv5 Based on Transformer Prediction Head for Object Detection on Drone-Captured Scenarios

Multi-task yolov5 with detection and segmentation based on yolov5

Yolov5-lite - Minimal PyTorch implementation of YOLOv5

Object tracking implemented with YOLOv4, DeepSort, and TensorFlow.

using yolox+deepsort for object-tracker

Vehicles Counting using YOLOv4 + DeepSORT + Flask + Ngrok

StrongSORT: Make DeepSORT Great Again

YoloV5 implemented by TensorFlow2 , with support for training, evaluation and inference.