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

Fixed image summaries

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

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

@ -192,7 +192,7 @@ def train_mnist(folding_id: int, inlier_classes: Sequence[int], total_classes: i
os.makedirs(directory)
comparison = k.concatenate([x[:64], x_decoded[:64]], axis=0)
grid = prepare_image(comparison.cpu(), nrow=64)
summary_ops_v2.image(name='reconstruction_' + str(epoch), tensor=grid, max_images=1)
summary_ops_v2.image(name='reconstruction_' + str(epoch), tensor=k.expand_dims(grid, axis=0), max_images=1)
from PIL import Image
filename = 'results' + str(inlier_classes[0]) + '/reconstruction_' + str(epoch) + '.png'
ndarr = grid.cpu().numpy()
@ -220,7 +220,7 @@ def train_mnist(folding_id: int, inlier_classes: Sequence[int], total_classes: i
directory = 'results' + str(inlier_classes[0])
os.makedirs(directory, exist_ok=True)
grid = prepare_image(resultsample)
summary_ops_v2.image(name='sample_' + str(epoch), tensor=grid, max_images=1)
summary_ops_v2.image(name='sample_' + str(epoch), tensor=k.expand_dims(grid, axis=0), max_images=1)
from PIL import Image
filename = 'results' + str(inlier_classes[0]) + '/sample_' + str(epoch) + '.png'
ndarr = grid.cpu().numpy()