vitis_inspect.VitisInspector.inspect_model - 3.5 简体中文

Vitis AI 用户指南 (UG1414)

Document ID
UG1414
Release Date
2023-09-28
Version
3.5 简体中文
此函数用于执行浮点模型检查:

VitisInspector.inspect_model(model, 
                             input_shape=None,
                             dump_model=True,
                             dump_model_file="inspect_model.h5",
                             plot=True,
                             plot_file="model.svg",
                             dump_results=True,
                             dump_results_file="inspect_results.txt",
                             verbose=0)

实参

model
tf.keras.Model 实例。表示要检查的浮点模型。浮点模型应包含具体的输入形状。以具体的输入形状来构建浮点模型,或者以 input_shape 实参调用 inspect_model。
input_shape
list(int) 或 list(list(int)),包含每个输入层的输入形状。如不设置,则使用输入层中的默认形状信息。使用多个输入的形状的列表,例如,inspect_model(model, input_shape=[224, 224, 3]) 或 inspect_model(model, input_shape=[[224, 224, 3], [64, 1]])。所有维度都应包含具体的值,且 batch_size 维度应省略。默认值为 None。
dump_model
bool。指示是否转储检查的模型并将模型保存到磁盘。默认值为 False。
dump_model_file
string。表示检查的模型文件的路径。默认值为“inspect_model.h5”。
plot
bool。表示是否绘制 graphviz 的模型检查结果并将图像保存到磁盘。如需将模型检查结果可视化,并提供某些修改提示,那么此项很有用。仅部分输出文件类型能显示提示,例如,.svg。默认值为 False。
plot_file
string。表示绘制模型时的模型图像文件的文件路径。默认值为 model.svg。
dump_results
bool。表示是否转储检查结果并将文本保存到磁盘。相比于屏幕日志记录,它能将更详细的逐层结果转储到文本文件。默认值为 False。
dump_results_file
string。表示检查结果文本文件的文件路径。默认值为“inspect_results.txt”。
verbose
int。表示日志记录的详细程度,verbose 值越高,显示的日志记录结果越详细。默认值为 0。