Я преобразовал свой файл yolo.weight в файл .onnx, и при преобразовании его в tenorrt я получил сообщение об ошибке
IndexError: Атрибут не найден: height_scale
Версия Tensorrt: 5.0.2.6 CUDA: 10.0
Ссылочная ссылка для преобразования: https://github.com/zombie0117/yolov3-tiny-onnx-TensorRT
Журнал ошибок:
Traceback (последний вызов был последним): файл "onnx_to_tensorrt.py", строка 226, в main () файле "onnx_to_tensorrt.py", строка 184, в основном с get_engine (onnx_file_path, batch_size, fp16_on, engine_file_path) в качестве engine, engine.create_execution_context () в качестве контекста: файл «onnx_to_tensorrt.py», строка 126, в get_engine возвращают build_ngine_tensto «on_inginet» на «file_nginert» на «file_enginet» на «file_nginer» на «file_enginet» на » строка 111 в build_engine parser.parse (model.read ()) IndexError: атрибут не найден: height_scale