From b4b066769f1883c69292017b8d88f0f3c2cf6e5a Mon Sep 17 00:00:00 2001 From: Haochen Wang Date: Fri, 10 Aug 2018 18:42:29 -0500 Subject: [PATCH] fix viz that produces large number of disk writes --- lib/utils/vis.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lib/utils/vis.py b/lib/utils/vis.py index db8676e6..85f03fdf 100644 --- a/lib/utils/vis.py +++ b/lib/utils/vis.py @@ -239,6 +239,6 @@ def vis_one_image( line, color=colors[len(kp_lines) + 1], linewidth=1.0, alpha=0.7) - output_name = os.path.basename(im_name) + '.' + ext - fig.savefig(os.path.join(output_dir, '{}'.format(output_name)), dpi=dpi) - plt.close('all') + output_name = os.path.basename(im_name) + '.' + ext + fig.savefig(os.path.join(output_dir, '{}'.format(output_name)), dpi=dpi) + plt.close('all')