修改模型输入数据格式

Python/YoloV5_infer_migraphx.py

@@ -38,7 +38,7 @@ class YOLOv5:
         # 执行推理
         print("Start to inference")
         start = time.time()
-        result = self.model.run({self.model.get_parameter_names()[0]: migraphx.argument(input_img)})
+        result = self.model.run({self.model.get_parameter_names()[0]: input_img})
         print('net forward time: {:.4f}'.format(time.time() - start))
         # 模型输出结果后处理
         boxes, scores, class_ids = self.process_output(result)