From ec6ea56c59f093729da5108b367fe83b5f110562 Mon Sep 17 00:00:00 2001 From: SwastiKh Date: Mon, 26 Oct 2020 23:00:40 +0530 Subject: [PATCH 1/4] config.py reduced min_after_dequeue slight modification in train.py --- config.py | 2 +- train.py | 8 ++++++++ 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/config.py b/config.py index 8903bd0..82f7f45 100644 --- a/config.py +++ b/config.py @@ -21,5 +21,5 @@ def __init__(self, **kwargs): self.checkpoint_basename = kwargs.get("checkpoint_basename", "layout") self.filenamequeue = kwargs.get("filenamequeue", os.path.join(self.data_dir, "dataset","layout_1205.tfrecords")) - self.min_after_dequeue = kwargs.get("min_after_dequeue", 5000) + self.min_after_dequeue = kwargs.get("min_after_dequeue", 500) self.num_threads = kwargs.get("num_threads", 4) diff --git a/train.py b/train.py index b503b49..bc386c0 100644 --- a/train.py +++ b/train.py @@ -1,5 +1,13 @@ import config import trainer_step +import tensorflow +from tensorflow.compat.v1 import ConfigProto +from tensorflow.compat.v1 import InteractiveSession + +config = ConfigProto() +config.gpu_options.allow_growth = True +session = InteractiveSession(config=config) + def main(): cfg = config.Config(filenamequeue="drive/My Drive/LayoutNet/dataset/layout_1205.tfrecords") From b1e68e4bb8b0a069c03801df84a8c20aad9c630b Mon Sep 17 00:00:00 2001 From: SwastiKh Date: Sat, 21 Nov 2020 08:12:16 +0530 Subject: [PATCH 2/4] trainer_step.py data path changed for trained model --- trainer_step.py | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/trainer_step.py b/trainer_step.py index 1f80a71..5fbe173 100644 --- a/trainer_step.py +++ b/trainer_step.py @@ -11,7 +11,7 @@ tf.compat.v1.disable_eager_execution() # slim = tf.contrib.slim -DATA_DIR = os.path.join('..','data') +DATA_DIR = os.path.join('..') class Trainer(object): def __init__(self, config): filenamequeue = tf.compat.v1.train.string_input_producer([config.filenamequeue]) @@ -58,7 +58,8 @@ def _build_model(self, filenamequeue, config): randomz = tf.random.normal([config.batch_size, config.z_dim]) # testing case testLayout, testImgfea, testSemvec, testTexfea = self.inputs(config) - randomz_val = np.load(os.path.join(DATA_DIR,'sample','noiseVector_128.npy')) + # randomz_val = np.load(os.path.join(DATA_DIR,'/drive/MyDrive/LayoutNetme/sample','noiseVector_128.npy')) + randomz_val = np.load('../drive/MyDrive/LayoutNetme/sample/noiseVector_128.npy') testSemvec, _ = model.embeddingSemvec(testSemvec, training_, reuse=True) testImgfea, _ = model.embeddingImg(testImgfea, training_, reuse=True) testTexfea, _ = model.embeddingTex(testTexfea, training_, reuse=True) @@ -223,8 +224,8 @@ def sample(self): return (inputdata + 1) / 2.0, (gen + 1) / 2.0, fea def testing(self): - new_saver = tf.compat.v1.train.import_meta_graph('./log/layoutNet-100.meta') - new_saver.restore(self.sess, './log/layoutNet-100') + new_saver = tf.compat.v1.train.import_meta_graph('/layoutNet-100.meta') + new_saver.restore(self.sess, '/layoutNet-100') gen, fea = self.sess.run([self.model["Gtest"], self.model["Etest"]], feed_dict={self.model["is_training"]: False}) @@ -232,12 +233,12 @@ def testing(self): return (gen + 1) / 2.0, fea def inputs(self, config): - layoutpath = DATA_DIR + '/sample/layout/' - imgfeapath = DATA_DIR + '/sample/visfea/' - texfeapath = DATA_DIR + '/sample/texfea/' - semvecpath = DATA_DIR + '/sample/semvec/' + layoutpath = DATA_DIR + '/drive/MyDrive/LayoutNetme/validation/layout/' + imgfeapath = DATA_DIR + '/drive/MyDrive/LayoutNetme/validation/visfea/' + texfeapath = DATA_DIR + '/drive/MyDrive/LayoutNetme/validation/texfea/' + semvecpath = DATA_DIR + '/drive/MyDrive/LayoutNetme/validation/semvec/' - f = open((DATA_DIR + '/sample/imgSel_128.txt'), 'r') + f = open((DATA_DIR + '/drive/MyDrive/LayoutNetme/validation/imgSel_60.txt'), 'r') name = f.read() namelist = name.split() n_samples = len(namelist) From 50bf1ab46e9dea30e41f372b52cf9808ada68114 Mon Sep 17 00:00:00 2001 From: SwastiKh Date: Sat, 21 Nov 2020 08:14:57 +0530 Subject: [PATCH 3/4] config.py changed argument back to 5000 --- config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/config.py b/config.py index b2966e0..6a1eec7 100644 --- a/config.py +++ b/config.py @@ -21,5 +21,5 @@ def __init__(self, **kwargs): self.checkpoint_basename = kwargs.get("checkpoint_basename", "layout") self.filenamequeue = kwargs.get("filenamequeue", os.path.join(self.data_dir, "dataset","layout_1205.tfrecords")) - self.min_after_dequeue = kwargs.get("min_after_dequeue", 500) + self.min_after_dequeue = kwargs.get("min_after_dequeue", 5000) self.num_threads = kwargs.get("num_threads", 4) From 820c0af059e1c51c1f7289be090b9efcb8a5e680 Mon Sep 17 00:00:00 2001 From: SwastiKh Date: Sat, 21 Nov 2020 08:17:57 +0530 Subject: [PATCH 4/4] reverted changes in train.py --- train.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/train.py b/train.py index 88358b8..79efe62 100644 --- a/train.py +++ b/train.py @@ -1,12 +1,5 @@ import config import trainer_step -import tensorflow -from tensorflow.compat.v1 import ConfigProto -from tensorflow.compat.v1 import InteractiveSession - -config = ConfigProto() -config.gpu_options.allow_growth = True -session = InteractiveSession(config=config) def main():