diff --git a/utils/utils.py b/utils/utils.py index 44cfd126..a6c15c6d 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -501,6 +501,7 @@ def plot_images(imgs, targets, fname='images.jpg'): plt.axis('off') fig.tight_layout() fig.savefig(fname, dpi=fig.dpi) + plt.close() def plot_results(start=0, stop=0): # from utils.utils import *; plot_results()