spacewalk01 / yolov9-bytetrack-tensorrt

Integration of YOLOv9 with ByteTracker
https://github.com/WongKinYiu/yolov9
75 stars 12 forks source link

Nothing happend after yolov9-bytetrack-trt.exe complete #3

Closed ffdalimarta closed 8 months ago

ffdalimarta commented 8 months ago

Hi Spacewalk, this is awesome. I follow the instruction to install and generate all requirements, and everything goes perfect until when I tried to : yolov9-bytetrack-trt.exe yolov9-c.engine switch1.mp4 in my console, it's nothing happened.

Did I missed something here?

PS: I've placed the video file in the same location as the .exe Screenshot 2024-02-28 114317

spacewalk01 commented 8 months ago

make sure that yolov9-c.engine and switch1.mp4 video files are present in Release directory. Or you can use absolute path for these files

ffdalimarta commented 8 months ago

I think I missed the yolov9-c.engine file here. Can you show me where can I obtain that file?

spacewalk01 commented 8 months ago

Sorry it wasn't in the instruction. So added it. You can build the engine using following command:

trtexec.exe --onnx=yolov9-c-converted.onnx --explicitBatch --saveEngine=yolov9-c.engine --fp16

trtexec is in the 'TensorRT root directory'/bin directory

ffdalimarta commented 8 months ago

thank you for the update. I follow your suggestion then using:

trtexec.exe --onnx=yolov9-c-converted.onnx --explicitBatch --saveEngine=yolov9-c.engine --fp16

but there is a problem in the generation of the engine. Did you ever encounter this problem below :

Screenshot 2024-02-28 122036

_PS: the required nvinfer_plugin.dll is in the TensorRT/lib folder, and I have added TensorRT/lib location to the PATH, but the error still persist_

spacewalk01 commented 8 months ago

Set tensorrt installation path in your envornment variable or copy dll to C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.x\bin or you can copy it to the folder you are running your exe TensorRT/bin and Your_project/builld/Release