From a48249554cbabaf797e7f7bdb4b5651f8d6fe461 Mon Sep 17 00:00:00 2001
From: Harry Stuart <42882697+HStuart18@users.noreply.github.com>
Date: Fri, 3 Jan 2020 12:24:29 +1100
Subject: [PATCH] updated .gitignore

---
 .gitignore         |  6 +++++-
 scripts/WGAN-GP.py | 10 ++++++++++
 2 files changed, 15 insertions(+), 1 deletion(-)

diff --git a/.gitignore b/.gitignore
index 4f21f62..d2afb40 100644
--- a/.gitignore
+++ b/.gitignore
@@ -242,4 +242,8 @@ paket-files/
 
 # JetBrains Rider
 .idea/
-*.sln.iml
\ No newline at end of file
+*.sln.iml
+
+# Custom
+media/
+logs/
\ No newline at end of file
diff --git a/scripts/WGAN-GP.py b/scripts/WGAN-GP.py
index 2bc9172..c9b953d 100644
--- a/scripts/WGAN-GP.py
+++ b/scripts/WGAN-GP.py
@@ -29,6 +29,16 @@ Fs = 16000
 DATA_DIR = r"D:\ML_Datasets\violin"
 INSTRUMENT = "violin"
 
+print("Creating necessary directories")
+
+paths = ["logs/train", 
+         f"models/{INSTRUMENT}/js",
+         f"output/{INSTRUMENT}",]
+
+for path in paths:
+    if not os.path.exists(os.path.join(os.getcwd(), path)):
+        os.makedirs(path)
+
 # Define class that contains GAN infrastructure
 class GAN:
     def __init__(self, model_dims=MODEL_DIMS, num_samples=NUM_SAMPLES,