From 151ade996065ad02c2d96b723c94589066491d54 Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Tue, 7 Jan 2020 16:51:38 +0100 Subject: upload bytesio object to cortex --- inversion/image_inversion_placeholder.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'inversion') diff --git a/inversion/image_inversion_placeholder.py b/inversion/image_inversion_placeholder.py index 64929cc..88af8ef 100644 --- a/inversion/image_inversion_placeholder.py +++ b/inversion/image_inversion_placeholder.py @@ -9,8 +9,8 @@ import itertools import numpy as np import os import params -import PIL import scipy +import pickle import sys import tensorflow as tf import tensorflow_probability as tfp @@ -485,6 +485,9 @@ for image_batch, label_batch in image_gen: out_labels[out_pos:out_pos+BATCH_SIZE] = label_batch out_err[out_pos:out_pos+BATCH_SIZE] = rec_err_batch out_pos += BATCH_SIZE + # upload image + # write encoding, latent to pkl file + if params.max_batches > 0 and (out_pos / BATCH_SIZE) >= params.max_batches: break -- cgit v1.2.3-70-g09d2