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