tensorrtx
tensorrtx copied to clipboard
widonws下yolov8模型生成wts转engine文件报错
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
请问是什么原因,应该如何解决呢