dbnet tensorboard可视化跟进训练过程

目录

1. 目的

2. 代码

3. 可视化结果


 

1. 目的

利用tensorboard实时的跟进训练情况,可视化当前的预测结果,和参与的数据增强后的训练数据情况。

2. 代码

if self.every_n_inner_iters(runner, self._print_interval_iter):
# loss
loss = runner.batch_iter_output['batch_loss']
prob_dice_loss = runner.batch_iter_output['prob_dice_loss']
thres_dice_loss = runner.batch_iter_output['thres_dice_loss']
thres_l1_loss = runner.batch_iter_out

你可能感兴趣的:(mmocr,OCR,深度学习)