shouxieai / hard_decode_trt

Yolov5 inference on NVDec hardware decoder

🤗 Enable AI with One Click!
The shouxieAI's official Bilbili （手写AI B站）：https://space.bilibili.com/1413433465
CV教程：https://github.com/shouxieai/A-series-of-CV
NLP教程：https://github.com/shouxieai/A-series-of-NLP

hard_decode_trt's Issues

Support Windows platform?

硬解码有内存泄漏，在centos上时，创建完后，进行销毁，会直接引起崩溃。

windows下不会崩溃，但是观察内存，发现每次创建后，销毁后，内存增大概8k.
而在centos上，直接线程中进行销毁时，直接崩溃了。

推理结果显示问题，实时硬解码推理需要实时显示，如何从worker中获取到原图的cv::Mat图像然后画图？

老师，我看目前这边的代码是commit完后，只出了个结果数据，然后推理显示是，又用opencvCapture读取了一遍视频，然后把结果显示和画到图上。但是实时推理需要实时显示，怎么从worker中实时返回画结果的图像和结果数据？
怎么弄核函数来画图，然后把结果数据画图，然后传出去，cv::Mat，显示？

CUDA Driver error

运行的时候出现下面的错误：
[2021-11-11 10:40:49][info][trt_infer.cpp:190]: 0.output : shape {16 x 25200 x 85}, Float32
[2021-11-11 10:40:49][error][cuda_tools.cpp:17]:CUDA Driver error cuvidCtxLockCreate(&m_ctxLock, cuContext) # (null), code = (null) [ -390026416 ] in file /workspace/hard_decode_trt/src/ffhdd/cuvid_decoder.cpp:95
[2021-11-11 10:40:49][error][app_yolo.cpp:134]:decoder create failed
[2021-11-11 10:40:49][info][yolo.cpp:288]:Engine destroy.
请问，这是驱动的问题吗，是否要更新驱动程序？

关于软硬解的速度差异

你好！我这边编译了项目，运行了示例程序的yolo检测，结果如下：

这里用的是默认的MP4文件，硬解速度只比软解快一点，差别不大；我改成了本地的RTSP视频流后结果如下：

这里我只解码了100帧就退出了，如图所示硬解速度比软解还慢一点，我这边用的是NVIDIA GeForce 3090的显卡，请问一下这个结果正常吗？或者能不能给出个参考数据？

Error Code2：Internal Error