Подача входных данных в модель onnx resnet18 (развертывание Resnet18 для обнаружения объектов в видеофайле) ⇐ Python

[Anonymous]

Я создал и сохранил обученную модель resnet18, используя код в github по этой ссылке.
код можно запустить, указав каталог обучения и тип сетевой модели. >
Модель resnet18.onnx выбрана и обучена для классификации 4 типов ячеек.
Для этого проекта я использую Nvidia jetson (ubuntu).
теперь мне нужно использовать сгенерированную обученную модель из приведенного выше кода (resnet18.onnx) для классификации объектов в видео, используя следующий фрагмент кода, где на обнаруженных объектах отображается небольшой прямоугольник и значение прогноза. ячейка в видео.
при запуске приведенного выше кода с помощью resnet18.onnx я получаю сообщение об ошибке:

Код: Выделить всё

confidences, boxes = ort_session.run(None, {input_name: img})

ort_session.run   expected 2 get 1

Какой второй вход ожидает модель onnx? Я знаю, что для классификации модели требуется только изображение, так какой второй входной сигнал необходим?

Подробнее здесь: https://stackoverflow.com/questions/622 ... etction-in