  import time
  import numpy
  from lib.training_data import TrainingDataGenerator, stack_images
  
  encoderH5 = '/encoder.h5'
  decoder_AH5 = '/decoder_A.h5'
  decoder_BH5 = '/decoder_B.h5'
  
  class ModelAE:
      def __init__(self, model_dir):
  
          self.model_dir = model_dir
 
</s> remove from lib.utils import get_image_paths
 </s> add from lib.utils import get_image_paths, get_folder </s> remove from lib.utils import BackgroundGenerator
 </s> add from lib.utils import BackgroundGenerator, get_folder </s> remove     verify_output = False
 </s> add 
encoderH5 = 'encoder.h5'
decoder_AH5 = 'decoder_A.h5'
decoder_BH5 = 'decoder_B.h5'

  import time
  import numpy
  from lib.training_data import TrainingDataGenerator, stack_images
  
encoderH5 = 'encoder.h5'
decoder_AH5 = 'decoder_A.h5'
decoder_BH5 = 'decoder_B.h5'
encoderH5 = 'encoder.h5'
decoder_AH5 = 'decoder_A.h5'
decoder_BH5 = 'decoder_B.h5'
encoderH5 = 'encoder.h5'
decoder_AH5 = 'decoder_A.h5'
decoder_BH5 = 'decoder_B.h5'
  
  class ModelAE:
      def __init__(self, model_dir):
  
          self.model_dir = model_dir
keep keep keep keep replace replace replace keep keep keep keep keep
 Changes I forgot to push :-/ (#136) 
lib/ModelAE.py