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