浏览代码

增加log日志

zhy 1 周之前
父节点
当前提交
f36c398602
共有 1 个文件被更改,包括 4 次插入1 次删除
  1. 4 1
      tests/verify_tool_accuracy_test.py

+ 4 - 1
tests/verify_tool_accuracy_test.py

@@ -72,4 +72,7 @@ if __name__ == '__main__':
             total += 1
             if str(verify_result) == args.except_result:
                 correct += 1
-        print(f"model_name: {model_dir}, accuracy: {correct * 100.0 / total}%")
+        print(f"共验证: {len(onnx_files)}个")
+        print(f"验证成功: {correct}个")
+        print("------------------准确率指标如下-------------------------")
+        print(f"模型名称: {model_dir}, 准确率: {correct * 100.0 / total}%")