widonws下yolov8模型生成wts转engine文件报错

[XuJWood]

Env

• GPU, e.g. 4070
• OS, e.g. Win10, etc.
• Cuda v11.8
• TensorRT 8.6.1

About this repo

• which branch/tag/commit are you using?
• which model? yolov5, retinaface?

yolov8-seg

Your problem

• what is your command? e.g. sudo ./yolov5 -s
• what's your output?
• what output do you expect?

我从官网下载的yolov8s-seg.pt,在tensorrt项目中利用get_wts.py生成yolov8s-seg.wts，然后在项目中生成engine文件时报错。
Loading weights: ../weights/yolov8s-seg.wts
[05/27/2024-11:09:22] [W] [TRT] The implicit batch dimension mode has been deprecated. Please create the network with NetworkDefinitionCreationFlag::kEXPLICIT_BATCH flag whenever possible.
[05/27/2024-11:09:22] [E] [TRT] 3: (Unnamed Layer* 0) [Convolution]:kernel weights has count 864 but 0 was expected
[05/27/2024-11:09:22] [E] [TRT] 4: (Unnamed Layer* 0) [Convolution]: count of 864 weights in kernel, but kernel dimensions (3,3) with 3 input channels, 0 output channels and 1 groups were specified. Expected Weights count is 3 * 33 * 0 / 1 = 0
[05/27/2024-11:09:22] [E] [TRT] 4: [convolutionNode.cpp::nvinfer1::builder::ConvolutionNode::computeOutputExtents::58] Error Code 4: Internal Error ((Unnamed Layer 0) [Convolution]: number of kernel weights does not match tensor dimensions)
[05/27/2024-11:09:22] [E] [TRT] 3: [network.cpp::nvinfer1::Network::addResize::1421] Error Code 3: API Usage Error (Parameter check failed at: network.cpp::nvinfer1::Network::addResize::1421, condition: input.getDimensions().nbDims > 0

请问是什么原因，应该如何解决呢

[wang-xinyu]

@lindsayshuo pls advice.

[XuJWood]

已经解决，是因为模型转化的时候 float gd = 0.33, gw = 0.50; int max_channels = 1024;没有修改，设置为0，所以通道不匹配，谢谢！