Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[Solution] TensorRT 8.0.1 engine for YOLOv4 in standalone mode #21

Open
frenky-strasak opened this issue Nov 24, 2021 · 0 comments
Open
Assignees

Comments

@frenky-strasak
Copy link

To compile the yolov4 models with batched NMS, I had to change this:

  1. comment builder->allowGPUFallback(true); in SampleYolo.cpp. According docs the allowGPUFallback was removed from TensorRT 8.0.1. If I understand well, I do not have to care about it in TensorRT 8.0.1. Am I right?
  2. comment all MYELIN_LIB and ENABLE_MYELIN, in Makefile.config to prevent error: /usr/bin/ld: cannot find -lmyelin because it seems to me that TensorRT 8.0.1 does not use it, but I found nothing about it. Do you have any idea?

If you do these 2 steps, then you can compile it and it works.
Good luck!

@frenky-strasak frenky-strasak changed the title TensorRT 8.0.1 engine for YOLOv4 in standalone mode [Solution]TensorRT 8.0.1 engine for YOLOv4 in standalone mode Nov 24, 2021
@frenky-strasak frenky-strasak changed the title [Solution]TensorRT 8.0.1 engine for YOLOv4 in standalone mode [Solution] TensorRT 8.0.1 engine for YOLOv4 in standalone mode Nov 24, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants