类似这个问题,
[Convolution]:kernel weights has count 3840 but 32640 was expected,生成engine模型失败
过程是生成->pt模型->导出wts->导出engine
原因是训练的pt模型的类别数量和导出时候用到的导出engine工具中的数量不一致导出的
解决方法:
1,训练模型修改:data->中的myvoc.yaml数量 80改为5
修改models->yolov5s.yalm中数量 80改为5
2,修改生成engine的代码数量,原来的6及其以前的版本是在yololayer.h里修改:constexpr static int kNumClass = 5;80修改为5,最新的7及master修改位置在config.h中要注意了
只要yolo训练和生成engine的数量一致就不会报这个错误了。