GithubHelp home page GithubHelp logo

yolov8_bytetrack's Introduction

yolov8_ByteTrack

一个很随意、简单的yolov8 bytetrack跟踪实现,目的是将yolov8推理和ByteTrack独立分离出来,摆脱屎山代码堆,方便代码修改和加入相应模块测试。

依赖

基本上就是yolov8的相关依赖,如果你已经用过yolov8,就用那个conda环境就好了,可能还有一些bytetrack的依赖,使用的过程中看着安装

运行

将权重和视频文件路径配置好。直接运行infer.py文件即可

代码没有做太多优化,改起来很方便。

Tensort实现 yolov8_ByteTrack

  1. 运行export-det.py 文件,得到outputs修改后的onnx文件
python export-det.py --weights yolov8n.pt --iou-thres 0.65 --conf-thres 0.25 --topk 100 --opset 11 --sim --input-shape 1 3 640 640 --device cuda:0
  1. 使用trtexec 将onnx转trt
./trtexec --onnx=yolov8n.onnx --saveEngine=yolov8n.engine --fp16

拷贝yolov8n.engine到tensorrt 实现的C++工程中

yolov8_ByteTrack_TensorRT

参考

YOLOv8-TensorRT

ByteTrack

yolov8_bytetrack's People

Contributors

1079863482 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.