|
@@ -98,6 +98,7 @@ if __name__ == '__main__':
|
|
|
# 打印模型层数信息
|
|
# 打印模型层数信息
|
|
|
total_layers, count = get_onnx_layer_info(onnx_file)
|
|
total_layers, count = get_onnx_layer_info(onnx_file)
|
|
|
print(f"ONNX模型层数统计（{onnx_file}）:")
|
|
print(f"ONNX模型层数统计（{onnx_file}）:")
|
|
|
|
|
+ print(f"过滤主要层：\"Conv\", \"BatchNormalization\", \"Gemm\", \"Relu\", \"MaxPool\", \"AveragePool\", \"Add\"")
|
|
|
print(f"模型层数: {count}, 所有算子节点: {total_layers}")
|
|
print(f"模型层数: {count}, 所有算子节点: {total_layers}")
|
|
|
|
|
|
|
|
# verify_result = verify_tool.label_verification(onnx_file)
|
|
# verify_result = verify_tool.label_verification(onnx_file)
|
