tensorrtx icon indicating copy to clipboard operation
tensorrtx copied to clipboard

yolov5 v6.0版本

Open lyc728 opened this issue 2 years ago • 6 comments

无法进行推理,使用代码无法得到推理结果,保存的图是原图

lyc728 avatar Apr 18 '22 07:04 lyc728

I found the reason, but don't know how to fix it. There is no 'model.24.anchor_grid' in trained model with yolov5-6.0, but it do exist in pretrained models, and exist in yolov5-5.0 trained models.

Wangwang99999 avatar Jun 16 '22 02:06 Wangwang99999

I found the reason, but don't know how to fix it. There is no 'model.24.anchor_grid' in trained model with yolov5-6.0, but it do exist in pretrained models, and exist in yolov5-5.0 trained models.

i got the same problem, yolov5s v6.1 , using official way to convert model: python export.py --weights xxx.pt --include engine onnx --imgsz 640 --device 0 --batch-size 4

after compile and using: ./yolov5 -d ../test/xxx.engine ../samples

i got no result(no bbox) on output images

dao027 avatar Jul 18 '22 09:07 dao027

Guys, you need to find out why the trained model doesn't contain anchor info.

wang-xinyu avatar Jul 19 '22 02:07 wang-xinyu

Guys, you need to find out why the trained model doesn't contain anchor info.

roger, i will check this

dao027 avatar Jul 19 '22 02:07 dao027

@dao027 Hei,I have the same problem , can you solved it?

hua1024 avatar Sep 13 '22 05:09 hua1024

negative, cant use engine file from yolov5 official convertion, u need to convert pth model to engine by way in this repo

dao027 avatar Sep 14 '22 08:09 dao027

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.

stale[bot] avatar Nov 14 '22 10:11 stale[bot]