update
This commit is contained in:
13
main.py
13
main.py
@@ -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()
|
||||
|
Reference in New Issue
Block a user