2martens/masterthesis
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Added step values to image summaries

Browse Source 
Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens
2019-02-08 12:41:10 +01:00
parent d84b062bf9
commit 682e11b435
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/twomartens/masterthesis/aae/train.py
View File

@ -192,7 +192,9 @@ def train_mnist(folding_id: int, inlier_classes: Sequence[int], total_classes: i
os.makedirs(directory) os.makedirs(directory)
comparison = k.concatenate([x[:64], x_decoded[:64]], axis=0) comparison = k.concatenate([x[:64], x_decoded[:64]], axis=0)
grid = prepare_image(comparison.cpu(), nrow=64) grid = prepare_image(comparison.cpu(), nrow=64)
summary_ops_v2.image(name='reconstruction_' + str(epoch), tensor=k.expand_dims(grid, axis=0), max_images=1) summary_ops_v2.image(name='reconstruction_' + str(epoch),
tensor=k.expand_dims(grid, axis=0), max_images=1,
step=global_step_decoder)
from PIL import Image from PIL import Image
filename = 'results' + str(inlier_classes[0]) + '/reconstruction_' + str(epoch) + '.png' filename = 'results' + str(inlier_classes[0]) + '/reconstruction_' + str(epoch) + '.png'
ndarr = grid.cpu().numpy() ndarr = grid.cpu().numpy()
@ -220,7 +222,8 @@ def train_mnist(folding_id: int, inlier_classes: Sequence[int], total_classes: i
directory = 'results' + str(inlier_classes[0]) directory = 'results' + str(inlier_classes[0])
os.makedirs(directory, exist_ok=True) os.makedirs(directory, exist_ok=True)
grid = prepare_image(resultsample) grid = prepare_image(resultsample)
summary_ops_v2.image(name='sample_' + str(epoch), tensor=k.expand_dims(grid, axis=0), max_images=1) summary_ops_v2.image(name='sample_' + str(epoch), tensor=k.expand_dims(grid, axis=0),
max_images=1, step=global_step_decoder)
from PIL import Image from PIL import Image
filename = 'results' + str(inlier_classes[0]) + '/sample_' + str(epoch) + '.png' filename = 'results' + str(inlier_classes[0]) + '/sample_' + str(epoch) + '.png'
ndarr = grid.cpu().numpy() ndarr = grid.cpu().numpy()
@ -411,7 +414,7 @@ def extract_batch(data: np.ndarray, it: int, batch_size: int) -> tfe.Variable:
if __name__ == "__main__": if __name__ == "__main__":
tf.enable_eager_execution() tf.enable_eager_execution()
inlier_classes = [0] inlier_classes = [0]
iteration = 1 iteration = 2
train_summary_writer = summary_ops_v2.create_file_writer( train_summary_writer = summary_ops_v2.create_file_writer(
'./summaries/train/number-' + str(inlier_classes[0]) + '/' + str(iteration)) './summaries/train/number-' + str(inlier_classes[0]) + '/' + str(iteration))
with train_summary_writer.as_default(), summary_ops_v2.always_record_summaries(): with train_summary_writer.as_default(), summary_ops_v2.always_record_summaries():