It don't work with yolov5-6.0

[Wangwang99999]

Env

• GPU, GTX1060
• OS, Win10
• Cuda version, 10.2
• TensorRT version, TensorRT-7.0.0.11_cuda10.2

About this repo

• which branch/tag/commit are you using? master
• which model? yolov5-6.0

Your problem

when I debug yololayer.cu, in line 'mKernelCount = vYoloKernel.size()', mKernelCount is zero, and there is no detected result. I compared with pretrained model and found the different: There is no 'model.24.anchor_grid' in trained model with yolov5-6.0, but it do exist in pretrained models, and exist in trained model with yolov5-5.0 How can I fix this problem?

[wang-xinyu]

Did you try the latest gen_wts.py?

[stale[bot]]

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.