def saveImage(data, filename): sd = Shared() sd.incFlag('safeExit') logger = sd.getLogger() src = (data * 255).astype(np.uint8) if len(src.shape) == 4: assert (src.shape[0]) == 1 h = src.shape[1] w = src.shape[2] c = src.shape[3] src = np.reshape(src, [h, w, c]) else: assert (len(src.shape) == 3) spm.toimage(src).save(filename) logger.debug('image saved to \'%s\'' % filename) sd.decFlag('safeExit')
def save(self, path, name, save_round): sd = Shared() sd.incFlag('safeExit') self.__logger.info('saving model[%s]...' % self.__name) try: os.makedirs(path) self.__logger.info('directory \'%s\' was made for model saving.' % path) except OSError: if not os.path.isdir(path): self.__logger.error( cprint('target path is not a directory', 'red')) return False saver = tf.train.Saver() saver.save(self.__sess, path + name) self.__logger.info('model[%s] saved.' % self.__name) sd.decFlag('safeExit') return True