From 8d3372100aa41d5071e772d26488660cbe73e76e Mon Sep 17 00:00:00 2001 From: Rodrigo Bermudez Schettino Date: Fri, 29 Nov 2019 19:47:45 +0100 Subject: [PATCH] gangogh: Generate 'generated' directory automatically --- gangogh.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/gangogh.py b/gangogh.py index d4c00e17c..a5181fc3d 100644 --- a/gangogh.py +++ b/gangogh.py @@ -1,4 +1,5 @@ import os, sys +from pathlib import Path sys.path.append(os.getcwd()) @@ -33,6 +34,8 @@ PREITERATIONS = 2000 # Number of preiteration training cycles to run lib.print_model_settings(locals().copy()) +# Ensure that directory exists where ground truth and plots will be saved to. +Path('generated').mkdir(parents=True, exist_ok=True) def GeneratorAndDiscriminator(): return kACGANGenerator, kACGANDiscriminator