def call(self, cfg, dataset, train_brick):
acc = train_brick.metric['train_accurency'].get_avg()
loss = train_brick.metric['train_loss'].get_avg()
message = f'[{train_brick.epoch}/{cfg.MAX_EPOCH} {train_brick.step}] acc={acc[0]}/{acc[1]} loss={loss[0]}/{loss[1]}'
Logger.debug(message,
show_type=Logger.LOG_STYLE.DEFAULT,
forground=Logger.LOG_FRONT_COLOR.GREEN,
background=Logger.LOG_BACK_COLOR.DEFAULT)
def call(self, cfg, dataset, train_brick):
message = f'[{train_brick.epoch}/{cfg.MAX_EPOCH} {train_brick.step}] start saving checkpoints...'
Logger.debug(message,
show_type=Logger.LOG_STYLE.DEFAULT,
forground=Logger.LOG_FRONT_COLOR.GREEN,
background=Logger.LOG_BACK_COLOR.DEFAULT)
## TODO
checkpoint_file = cfg.save_checkpoint_prefix + '_' + str(
train_brick.epoch) + '_' + str(train_brick.step)
checkpoint_file_path = os.path.join(cfg.save_checkpoint_path,
checkpoint_file)
train_brick.model.save_weights(checkpoint_file_path)
message = f'[{train_brick.epoch}/{cfg.MAX_EPOCH} {train_brick.step}] complete save checkpoints.'
Logger.debug(message,
show_type=Logger.LOG_STYLE.DEFAULT,
forground=Logger.LOG_FRONT_COLOR.GREEN,
background=Logger.LOG_BACK_COLOR.DEFAULT)
def debug(msg: str):
"""
log Debug
:param msg:
"""
Logger.debug(msg)
class DownloadMacro:
__IMAGES = [
'click_button.png', 'download.png', 'next_button.png',
'return_back.png', 'next_page.png'
]
__PAGE_COUNT = 5 #크롤링할 페이지의 개수
__INDEX_COUNT = 25 #한 페이지 내 게임의 개수
__USER_RESOLUTION = 62 #사용자 해상도
def __init__(self):
self.logger = Logger('DownloadMacro')
self.logger.debug("init 완료.")
def img_click(self, img):
self.logger.debug("img_click: {} start".format(img))
exit_count = 0 #현재 페이지 내에 img가 존재하지 않는 경우 exit하기 위해 사용.
while True:
button = m.locateCenterOnScreen(
'./img/' + img) #페이지 안에 존재하는 img를 찾아 위치를 가져온다.
if button != None:
self.logger.debug("img_click: {} button click".format(img))
m.moveTo(button.x, button.y, 1)
time.sleep(1)
m.click()
break
else:
if exit_count > 10:
self.logger.debug(
"img_click: {} button is not exist".format(img))
break
else:
exit_count += 1
m.press('pgdn')
time.sleep(1)
def Macro(self):
for page in range(DownloadMacro.__PAGE_COUNT):
self.logger.debug("page: {} / {}".format(
page + 1, DownloadMacro.__PAGE_COUNT))
target_position = (0, 0)
for index in range(DownloadMacro.__INDEX_COUNT):
self.logger.debug("index: {} / {}".format(
index + 1, DownloadMacro.__INDEX_COUNT))
time.sleep(5)
if not index:
self.img_click(DownloadMacro.__IMAGES[0])
target_position = m.position()
else:
m.moveTo(target_position.x, target_position.y, 1)
time.sleep(1)
m.scroll(-DownloadMacro.__USER_RESOLUTION
) #사용자 환경에 따라 스크롤 길이 조절
time.sleep(1)
m.click()
time.sleep(2)
self.img_click(DownloadMacro.__IMAGES[1])
time.sleep(10)
self.img_click(DownloadMacro.__IMAGES[2])
m.click()
time.sleep(15)
self.img_click(DownloadMacro.__IMAGES[3])
time.sleep(1)
self.img_click(DownloadMacro.__IMAGES[4])