我已经使用此链接中 github 中的代码构建并保存了一个训练有素的 resnet18 模型
代码可以通过指定训练目录和网络模型类型来运行。
选择并训练模型 resnet18.onnx 对 4 种类型的细胞进行分类。
我正在为这个项目使用 Nvidia jetson (ubuntu)。
现在我需要使用从上面的代码（resnet18.onnx）生成的训练模型使用以下代码片段对视频中的对象进行分类，其中一个小框和预测值显示在视频中检测到的单元格上。
当我使用 resnet18.onnx 运行上述代码时收到的错误消息是
置信度，框 = ort_session.run（无，{input_name：img}）
ort_session.run 预期 2 得到 1
onnx 模型期望的第二个输入是什么（我知道模型只需要图像来对其进行分类，所以需要的第二个输入是什么）