diff options
Diffstat (limited to 'inversion/image_inversion_placeholder.py')
| -rw-r--r-- | inversion/image_inversion_placeholder.py | 5 |
1 files changed, 4 insertions, 1 deletions
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 |