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

set xlim and ylim by parameter

Browse Source
This commit is contained in:
mlxu995 2021-12-16 14:57:41 +08:00
parent 112003dd2d
commit 831ba946d6
2 changed files with 34 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
examples/hey_snips/s0/run.sh
View File

@ -4,7 +4,7 @@
. ./path.sh . ./path.sh
stage=0 stage=3
stop_stage=4 stop_stage=4
num_keywords=1 num_keywords=1
@ -14,7 +14,7 @@ norm_var=true
gpus="0" gpus="0"
checkpoint= checkpoint=
dir=exp/ds_tcn dir=exp/ds_tcn_specargument
num_average=30 num_average=30
score_checkpoint=$dir/avg_${num_average}.pt score_checkpoint=$dir/avg_${num_average}.pt
@ -109,7 +109,11 @@ if [ ${stage} -le 3 ] && [ ${stop_stage} -ge 3 ]; then
python kws/bin/plot_det_curve.py \ python kws/bin/plot_det_curve.py \
--keywords 'Hey_Snips' \ --keywords 'Hey_Snips' \
--stats_dir $result_dir \ --stats_dir $result_dir \
--figure_file $result_dir/det.png --figure_file $result_dir/det.png \
--xlim 10 \
--x_step 2 \
--ylim 10 \
--y_step 2
fi fi

36
kws/bin/plot_det_curve.py
View File

@ -25,16 +25,22 @@ def load_stats_file(stats_file):
for line in fin: for line in fin:
arr = line.strip().split() arr = line.strip().split()
threshold, fa_per_hour, frr = arr threshold, fa_per_hour, frr = arr
values.append([float(fa_per_hour), float(frr)]) values.append([float(fa_per_hour), float(frr) * 100])
values.reverse() values.reverse()
return np.array(values) return np.array(values)
def plot_det_curve(keywords, stats_dir, figure_file): def plot_det_curve(
keywords,
stats_dir,
figure_file,
xlim,
x_step,
ylim,
y_step):
plt.figure(dpi=200) plt.figure(dpi=200)
plt.rcParams['xtick.direction'] = 'in' plt.rcParams['xtick.direction'] = 'in'
plt.rcParams['ytick.direction'] = 'in' plt.rcParams['ytick.direction'] = 'in'
plt.rcParams['text.usetex'] = True
plt.rcParams['font.size'] = 12 plt.rcParams['font.size'] = 12
for index, keyword in enumerate(keywords): for index, keyword in enumerate(keywords):
@ -42,11 +48,10 @@ def plot_det_curve(keywords, stats_dir, figure_file):
values = load_stats_file(stats_file) values = load_stats_file(stats_file)
plt.plot(values[:, 0], values[:, 1], label=keyword) plt.plot(values[:, 0], values[:, 1], label=keyword)
plt.xlim([0, 5]) plt.xlim([0, xlim])
plt.ylim([0, 0.35]) plt.ylim([0, ylim])
plt.xticks([0, 1, 2, 3, 4, 5], ['0', '1', '2', '3', '4', '5']) plt.xticks(range(0, xlim + x_step, x_step))
plt.yticks([0, 0.05, 0.10, 0.15, 0.20, 0.25, 0.30, 0.35], plt.yticks(range(0, ylim + y_step, y_step))
['0', '5', '10', '15', '20', '25', '30', '35'])
plt.xlabel('False Alarm Per Hour') plt.xlabel('False Alarm Per Hour')
plt.ylabel('False Rejection Rate (\\%)') plt.ylabel('False Rejection Rate (\\%)')
plt.grid(linestyle='--') plt.grid(linestyle='--')
@ -67,8 +72,21 @@ if __name__ == '__main__':
'--figure_file', '--figure_file',
required=True, required=True,
help='path to save det curve') help='path to save det curve')
parser.add_argument('--xlim', required=True, help='range of X-axis')
parser.add_argument('--x_step', required=True, help='step of X-axis')
parser.add_argument('--ylim', required=True, help='range of Y-axis')
parser.add_argument('--y_step', required=True, help='step of Y-axis')
args = parser.parse_args() args = parser.parse_args()
keywords = args.keywords.strip().split(', ') keywords = args.keywords.strip().split(', ')
plot_det_curve(keywords, args.stats_dir, args.figure_file) xlim, x_step, ylim, y_step = map(
int, [args.xlim, args.x_step, args.ylim, args.y_step])
plot_det_curve(
keywords,
args.stats_dir,
args.figure_file,
xlim,
x_step,
ylim,
y_step)