fpga.love

EN App

知识库›AMD 社区问答速查›Vitis AI编译报输入层形状无效如何解决

Vitis AI编译报输入层形状无效如何解决

ILA 在线调试已闭环原帖 9 个回答 · 489 次浏览|器件：Vitis AI（错误码：AssertionError: [ERROR] Invalid shape of input layer: shape:）

一句话结论：编译时显式指定input_shape的NHWC四维值即可避免该错误

现象

编译ML模型时报AssertionError: [ERROR] Invalid shape of input layer: shape:

根因

Vitis AI编译器要求输入形状为(N,H,W,C)，NHWC参数不能为None，缺失时即报该错误

可尝试的处理步骤

在编译命令中添加参数 --options "{'mode':'normal','save_kernel':'', 'input_shape':'1,416,416,3'}" 后重新运行
参考vai_c_tensorflow命令示例，TensorFlow2、PyTorch等可类似处理

本文基于 AMD 客户社区（原 Xilinx 论坛）真实问答整理为中文要点，非原文转载；技术细节以原帖与 AMD 官方文档为准。
原帖：https://adaptivesupport.amd.com/s/question/0D54U00008FKXGeSAP/-info-post-resolving-compilation-error-invalid-s

相关资料

Vivado2015.2添加ILA后报DRC RTSTAT-6局部冲突 Vivado JTAG配置报错但FPGA编程成功 ISE中如何将启动时钟改为JTAG CLK？

相关标签：Vitis AI · compilation · input_shape · NHWC

就这个问题继续问 AI → ← 返回 AMD 社区问答速查