python - 我正在使用 Yolov5 预训练模型，我已经成功创建了 yolov5s。引擎文件，但 yolov5_trt.py 文件中有问题。核心转储

Question

我遵循所有提到的指令（https://github.com/wang-xinyu/tensorrtx/tree/master/yolov5）。运行 python yolov5_trt.py 文件时出错。

python: engine.cpp:1104: bool nvinfer1::rt::Engine::deserialize(const void*, std::size_t, nvinfer1::IGpuAllocator&, nvinfer1::IPluginFactory*): 断言`size >= bsize && "不匹配在分配的内存大小和序列化引擎的预期大小之间。"'失败。中止（核心转储）

TensorRT-5.1.5 操作系统：Ubuntu 16.04 torch：1.6.0+cu101 torchvision：0.7.0+cu101 cuda：10.1 python：3.6

谢谢你...

Answer 1

确保在 yololayer.h 中有正确数量的类