Yolo V5 转tensorrt

一、 在转tensorrt之前,请先确认你下载的yolo v5的版本及对应权重:

这个一定要注意,因为有的童鞋上来就转,如果成功了,那还好,成功不了,首先就要想到这个问题,就是yolo v5版本不对,导致转tensorrt不成功。自己可用这个版本的模型训练1个epoch,然后再转一下。如果转成功了,再去大规模的训练。

在官方地址:

https://github.com/ultralytics/yolov5

中有个master下有个Tags,选择v6.0版本

Yolo V5 转tensorrt_第1张图片

 2. 点击v6.0后下面给出这个版本的代码以及权重,去下面找它的权重

你可能感兴趣的:(深度学习,神经网络,pytorch)