0

我遵循所有提到的指令(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

谢谢你...

4

1 回答 1

0

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

于 2021-01-10T12:33:48.537 回答