YoLov5-TensorRT-NMS icon indicating copy to clipboard operation
YoLov5-TensorRT-NMS copied to clipboard

Published 20 hours ago • verified publisher icon upczww

Can you tell us which yolov5 version do you use? Does it surpport yolov5-v5.0???

Open sean-wade opened this issue 3 years ago • 5 comments

Thanks for replying ~

sean-wade avatar Jul 27 '21 14:07 sean-wade

yolov5 3.1

upczww avatar Jul 27 '21 16:07 upczww

[E] [TRT] (Unnamed Layer* 18) [Convolution]: kernel weights has count 3072 but 1024 was expected

My class nums is modified to 1, and when I convert 'wts' to 'engine', this error occurs...

Can you please tell me where is wrong... Thanks~

sean-wade avatar Aug 19 '21 03:08 sean-wade

Have you changed the model tpye? s/m/l/x?

upczww avatar Aug 19 '21 03:08 upczww

Have you changed the model tpye? s/m/l/x?

yes, I've changed it...

sean-wade avatar Aug 19 '21 03:08 sean-wade

update to 5.0

upczww avatar Sep 30 '21 08:09 upczww