修改准确率测试脚本

tests/verify_tool_accuracy_test.py

@@ -36,26 +36,27 @@ def filter_model_dirs(model_dir, targets):
 if __name__ == '__main__':
     parser = argparse.ArgumentParser(description='模型标签验证准确率验证脚本')
     parser.add_argument('--target_dir', default="origin_models", type=str, help='模型文件存放根目录，支持子文件夹递归处理')
-    parser.add_argument('--model_type', default=None, type=str, help='按照模型分类过滤，用于区分是目标检测模型还是图像分类模型，可选参数：classification、objection_detect')
-    parser.add_argument('--model_value', default=None, type=str, help='按照模型名称过滤，可选参数：alexnet、googlenet、resnet、vgg16、ssd、yolox、rcnn')
+    # parser.add_argument('--model_type', default=None, type=str, help='按照模型分类过滤，用于区分是目标检测模型还是图像分类模型，可选参数：classification、objection_detect')
+    # parser.add_argument('--model_value', default=None, type=str, help='按照模型名称过滤，可选参数：alexnet、googlenet、resnet、vgg16、ssd、yolox、rcnn')
     parser.add_argument('--model_file_filter', default=None, type=str, help='按照模型文件名过滤, 比如剪枝模型文件名存在prune。默认为None')
     parser.add_argument('--except_result', default=None, type=str, help='模型推理预期结果。默认为None')
 
     args, _ = parser.parse_known_args()
     if args.target_dir is None:
         raise Exception("模型目录参数不可为空")
-    if args.model_type is None:
-        raise Exception("模型类型参数不可为空")
+    # if args.model_type is None:
+    #     raise Exception("模型类型参数不可为空")
     if args.except_result is None:
         raise Exception("模型推理预期结果不可为空")
 
     # 获取所有模型目录信息
-    model_dirs = [item for item in os.listdir(args.target_dir) if os.path.isdir(os.path.join(args.target_dir, item))]
-    if args.model_type:
-        filter_models = model_types[args.model_type]
-        model_dirs = [item for item in model_dirs if filter_model_dirs(item, filter_models)]
-    if args.model_value:
-        model_dirs = [item for item in model_dirs if args.model_value.lower() in item.lower()]
+    # model_dirs = [item for item in os.listdir(args.target_dir) if os.path.isdir(os.path.join(args.target_dir, item))]
+    # if args.model_type:
+    #     filter_models = model_types[args.model_type]
+    #     model_dirs = [item for item in model_dirs if filter_model_dirs(item, filter_models)]
+    # if args.model_value:
+    #     model_dirs = [item for item in model_dirs if args.model_value.lower() in item.lower()]
+    model_dirs = [args.target_dir]
 
     # 遍历符合条件的模型目录列表，进行标签提取检测，并记录准确率
     for model_dir in model_dirs: