OLD | NEW |
1 #!/usr/bin/env python | 1 #!/usr/bin/env python |
2 # Copyright (c) 2017 The WebRTC project authors. All Rights Reserved. | 2 # Copyright (c) 2017 The WebRTC project authors. All Rights Reserved. |
3 # | 3 # |
4 # Use of this source code is governed by a BSD-style license | 4 # Use of this source code is governed by a BSD-style license |
5 # that can be found in the LICENSE file in the root of the source | 5 # that can be found in the LICENSE file in the root of the source |
6 # tree. An additional intellectual property rights grant can be found | 6 # tree. An additional intellectual property rights grant can be found |
7 # in the file PATENTS. All contributing project authors may | 7 # in the file PATENTS. All contributing project authors may |
8 # be found in the AUTHORS file in the root of the source tree. | 8 # be found in the AUTHORS file in the root of the source tree. |
9 | 9 |
10 """Perform APM module quality assessment on one or more input files using one or | 10 """Perform APM module quality assessment on one or more input files using one or |
(...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
56 help='custom list of evaluation scores to use', | 56 help='custom list of evaluation scores to use', |
57 choices=_EVAL_SCORE_WORKER_NAMES, | 57 choices=_EVAL_SCORE_WORKER_NAMES, |
58 default=_EVAL_SCORE_WORKER_NAMES) | 58 default=_EVAL_SCORE_WORKER_NAMES) |
59 | 59 |
60 parser.add_argument('-o', '--output_dir', required=False, | 60 parser.add_argument('-o', '--output_dir', required=False, |
61 help=('base path to the output directory in which the ' | 61 help=('base path to the output directory in which the ' |
62 'output wav files and the evaluation outcomes ' | 62 'output wav files and the evaluation outcomes ' |
63 'are saved'), | 63 'are saved'), |
64 default='output') | 64 default='output') |
65 | 65 |
| 66 parser.add_argument('--polqa_path', required=True, |
| 67 help='path to the POLQA tool') |
| 68 |
| 69 parser.add_argument('--air_db_path', required=True, |
| 70 help='path to the Aechen IR database') |
| 71 |
66 return parser | 72 return parser |
67 | 73 |
68 | 74 |
69 def main(): | 75 def main(): |
70 # TODO(alessiob): level = logging.INFO once debugged. | 76 # TODO(alessiob): level = logging.INFO once debugged. |
71 logging.basicConfig(level=logging.DEBUG) | 77 logging.basicConfig(level=logging.DEBUG) |
72 | 78 |
73 parser = _instance_arguments_parser() | 79 parser = _instance_arguments_parser() |
74 args = parser.parse_args() | 80 args = parser.parse_args() |
75 | 81 |
76 simulator = simulation.ApmModuleSimulator() | 82 simulator = simulation.ApmModuleSimulator( |
| 83 aechen_ir_database_path=args.air_db_path, |
| 84 polqa_tool_path=args.polqa_path) |
77 simulator.run( | 85 simulator.run( |
78 config_filepaths=args.config_files, | 86 config_filepaths=args.config_files, |
79 input_filepaths=args.input_files, | 87 input_filepaths=args.input_files, |
80 noise_generator_names=args.noise_generators, | 88 noise_generator_names=args.noise_generators, |
81 eval_score_names=args.eval_scores, | 89 eval_score_names=args.eval_scores, |
82 output_dir=args.output_dir) | 90 output_dir=args.output_dir) |
83 | 91 |
84 sys.exit(0) | 92 sys.exit(0) |
85 | 93 |
86 | 94 |
87 if __name__ == '__main__': | 95 if __name__ == '__main__': |
88 main() | 96 main() |
OLD | NEW |