修改模型输入数据格式

ce8766c6 · Your Name · 4d3d722b · ce8766c6
Commit ce8766c6 authored Jun 02, 2023 by Your Name
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

Python/RetinaFace_infer_migraphx.py Python/RetinaFace_infer_migraphx.py +1 -1

No files found.
--- a/Python/RetinaFace_infer_migraphx.py
+++ b/Python/RetinaFace_infer_migraphx.py
@@ -40,7 +40,7 @@ def migraphx_run(model,cpu,data_tensor):
        img_data[i, :, :, :] = data_numpy[i, :, :, :]

    # 执行推理
-    result = model.run({model.get_parameter_names()[0]: migraphx.argument(img_data)})
+    result = model.run({model.get_parameter_names()[0]: img_data})

    # 将结果转换为tensor
    result0=torch.from_numpy(np.array(result[0], copy=False)).to(device)