| import os
|
| import sys
|
|
|
| from tensorboardX import SummaryWriter
|
| from datetime import datetime
|
|
|
|
|
| class Logger(object):
|
| def __init__(self, val=True, filename="print.log"):
|
| self.Time = datetime.now().strftime('%Y-%m-%d_%H%M')
|
| self.path = 'output/' + self.Time
|
| self.log_filename = filename
|
| os.makedirs(self.path) if os.path.exists(self.path) is False else None
|
| self.run_path = '{}/{}'.format(self.path, 'tb')
|
|
|
|
|
| self.terminal = sys.stdout
|
| self.terminal.write(self.path)
|
|
|
|
|
| self.train_writer = None
|
| self.val_writer = None
|
| self.tensorboard_init(val)
|
|
|
| def printlog(self, message):
|
| message = str(message)
|
| self.terminal.write(message + '\n')
|
|
|
| log = open(os.path.join(self.path, self.log_filename), "a", encoding='utf8', )
|
| log.write(message + '\n')
|
| log.close()
|
|
|
| def tensorboard_init(self, val=True):
|
| if val:
|
| self.train_writer = SummaryWriter(self.run_path+'/train')
|
| self.val_writer = SummaryWriter(self.run_path+'/val')
|
| else:
|
| self.train_writer = SummaryWriter(self.run_path)
|
|
|
| def get_path(self):
|
| return self.path
|
|
|
