Я использую предобученную модель 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*): Assertion `size> = bsize &&" Mismatch между размером выделенной памяти и ожидаемым размером сериализованного движка."'не удалось. Прервано (ядро выгружено)

TensorRT-5.1.5os: Ubuntu 16.04 torch: 1.6.0+cu101torchvision: 0.7.0+cu101cuda: 10.1 python: 3.6

Благодарю вас...

Answer 1

Убедитесь, что у вас правильное количество классов в yololayer.h