This commit is contained in:
2025-09-04 11:06:09 +00:00
parent d6ae9a227f
commit 44febd7d2e
2 changed files with 91 additions and 3 deletions

13
main.py
View File

@@ -125,8 +125,15 @@ Examples:
logger.info(f"Output directory: {args.output_dir}")
# Determine which metrics to enable
enable_deqa = args.enable_deqa and not args.deqa_only
enable_traditional = args.enable_traditional and not args.deqa_only
if args.deqa_only:
# DeQA only: disable all other metric groups explicitly
enable_deqa = True
enable_traditional = False
enable_pyiqa_flag = False
else:
enable_deqa = args.enable_deqa
enable_traditional = args.enable_traditional
enable_pyiqa_flag = args.enable_pyiqa
# User-requested 20 metrics (NR + FR) from PyIQA
selected_top20 = [
@@ -142,7 +149,7 @@ Examples:
args.output_dir,
enable_deqa=enable_deqa,
enable_traditional=enable_traditional,
enable_pyiqa=args.enable_pyiqa,
enable_pyiqa=enable_pyiqa_flag,
pyiqa_selected_metrics=(selected_top20 if args.pyiqa_top20 else None)
)
results, report = analyzer.run_analysis()