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
