AI/wekws
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

format

Browse Source
This commit is contained in:
mlxu995 2021-12-16 13:08:23 +08:00
parent 8383ae2f90
commit 5f50d01ae3
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
kws/bin/plot_det_curve.py
View File

@ -56,10 +56,15 @@ def plot_det_curve(keywords, stats_dir, figure_file):
if __name__ == '__main__': if __name__ == '__main__':
parser = argparse.ArgumentParser(description='plot det curve') parser = argparse.ArgumentParser(description='plot det curve')
parser.add_argument('--keywords', required=True, parser.add_argument(
'--keywords',
required=True,
help='keywords, must in the same order as in "dict/words.txt" separated by ", "') help='keywords, must in the same order as in "dict/words.txt" separated by ", "')
parser.add_argument('--stats_dir', required=True, help='dir of stats files') parser.add_argument('--stats_dir', required=True, help='dir of stats files')
parser.add_argument('--figure_file', required=True, help='path to save det curve') parser.add_argument(
'--figure_file',
required=True,
help='path to save det curve')
args = parser.parse_args() args = parser.parse_args()