GithubHelp home page GithubHelp logo

Comments (9)

lencm avatar lencm commented on June 15, 2024 1

I have uploaded this 'libmyplugins.so' file, please test again.

环境为Ubuntu18.04 with CUDA 10.2 and cuDNN 8.0.0 TensorRT jetpack自带 PyTorch 1.8.0 and TorchVision 0.9.0 OpenCV-Python 4.5 pycuda 2019.1

你在Jetson板子上运行的吧?libmyplugins.so得重新编译生成

我也遇到同样的问题,非常好的项目,请问如何编译libmyplugins.so

build tensorrtx / yolov5 and generate ***.engine

cd {tensorrtx}/yolov5/

update CLASS_NUM in yololayer.h if your model is trained on custom dataset

mkdir build
cd build
cp {ultralytics}/yolov5/yolov5s.wts {tensorrtx}/yolov5/build
cmake ..
make
没记错的话是这个编译完会有,按流程来就行

from yolov5_deepsort_tensorrt.

cong avatar cong commented on June 15, 2024

I have uploaded this 'libmyplugins.so' file, please test again.

from yolov5_deepsort_tensorrt.

lencm avatar lencm commented on June 15, 2024

I have uploaded this 'libmyplugins.so' file, please test again.

我也遇到了这个问题,libmyplugins.so已经下载了,路径也没问题,不知道是不是环境哪里有问题呢?

from yolov5_deepsort_tensorrt.

lencm avatar lencm commented on June 15, 2024

I have uploaded this 'libmyplugins.so' file, please test again.

环境为Ubuntu18.04 with CUDA 10.2 and cuDNN 8.0.0
TensorRT jetpack自带
PyTorch 1.8.0 and TorchVision 0.9.0
OpenCV-Python 4.5
pycuda 2019.1

from yolov5_deepsort_tensorrt.

cong avatar cong commented on June 15, 2024

I have uploaded this 'libmyplugins.so' file, please test again.

环境为Ubuntu18.04 with CUDA 10.2 and cuDNN 8.0.0 TensorRT jetpack自带 PyTorch 1.8.0 and TorchVision 0.9.0 OpenCV-Python 4.5 pycuda 2019.1

你在Jetson板子上运行的吧?libmyplugins.so得重新编译生成

from yolov5_deepsort_tensorrt.

lencm avatar lencm commented on June 15, 2024

嗯嗯,使用自己生成的文件后跑通了。目前只能用CPU跑,速度在0.5fps,用CUDA的话会报错,单独测试cuda没问题,重装了pytorch还是不行。不知道哪里有问题
Traceback (most recent call last):
File "demo_trt.py", line 60, in
detect(video_path, engine_file_path)
File "demo_trt.py", line 31, in detect
bboxes = detector.detect(img)
File "/home/nvidia/yolov5_deepsort_tensorrt/detector_trt.py", line 120,
in detect
results_trt= self.post_process_new(trt_outputs, origin_h, origin_w)
File "/home/nvidia/yolov5_deepsort_tensorrt/detector_trt.py", line 255,
in post_process_new
pred = torch.Tensor(pred).cuda()
File "/home/nvidia/.local/lib/python3.6/site-packages/torch/cuda/ini
t
.py", line 164, in _lazy_init
raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled

from yolov5_deepsort_tensorrt.

lencm avatar lencm commented on June 15, 2024

I have uploaded this 'libmyplugins.so' file, please test again.

环境为Ubuntu18.04 with CUDA 10.2 and cuDNN 8.0.0 TensorRT jetpack自带 PyTorch 1.8.0 and TorchVision 0.9.0 OpenCV-Python 4.5 pycuda 2019.1

你在Jetson板子上运行的吧?libmyplugins.so得重新编译生成

问题已解决,jetson有专用的pytorch和torchvision包,需要去nvidia官网安装。目前GPU已跑通,但是NX上的帧率是2fps左右,和您说的6fps有些差距,不知道是为什么,您能提供一些思路吗?

from yolov5_deepsort_tensorrt.

cong avatar cong commented on June 15, 2024

I have uploaded this 'libmyplugins.so' file, please test again.

环境为Ubuntu18.04 with CUDA 10.2 and cuDNN 8.0.0 TensorRT jetpack自带 PyTorch 1.8.0 and TorchVision 0.9.0 OpenCV-Python 4.5 pycuda 2019.1

你在Jetson板子上运行的吧?libmyplugins.so得重新编译生成

问题已解决,jetson有专用的pytorch和torchvision包,需要去nvidia官网安装。目前GPU已跑通,但是NX上的帧率是2fps左右,和您说的6fps有些差距,不知道是为什么,您能提供一些思路吗?

目标的数量一定程度上会影响速度。我过几天有空会更新一版代码,可能会稍微提速一下,具体提升多少还没有空测试过。

from yolov5_deepsort_tensorrt.

43998213 avatar 43998213 commented on June 15, 2024

I have uploaded this 'libmyplugins.so' file, please test again.

环境为Ubuntu18.04 with CUDA 10.2 and cuDNN 8.0.0 TensorRT jetpack自带 PyTorch 1.8.0 and TorchVision 0.9.0 OpenCV-Python 4.5 pycuda 2019.1

你在Jetson板子上运行的吧?libmyplugins.so得重新编译生成

我也遇到同样的问题,非常好的项目,请问如何编译libmyplugins.so

from yolov5_deepsort_tensorrt.

Related Issues (19)

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.