YOLOv3代码阅读笔记之logger.py（第三篇）

对YOLOv3进行阅读，因为本人是小白，可能理解不到位的地方，请见谅。源码fork自eriklindernoren/PyTorch-YOLOv3，如需下载，请移步github,自行搜索。
logger.py通过引入tensorflow来记录

import tensorflow as tf#引入tensorflow，此文件使用tensorflow来记录


class Logger(object):
    def __init__(self, log_dir):#log_dir是日志的路径
        """Create a summary writer logging to log_dir."""
        self.writer = tf.summary.FileWriter(log_dir)#创建一个summary writer
        #由于版本问题，tf.summary.FileWriter可能会报错，改为tf.compat.v1.summary.FileWriter

    def scalar_summary(self, tag, value, step):#记录a scalar variable
        """Log a scalar variable."""
        summary = tf.Summary(value=[tf.Summary.Value(tag=tag, simple_value=value)])
        self.writer.add_summary(summary, step)
        #由于版本问题，tf.summary.FileWriter可能会报错，改为tf.compat.v1.summary.FileWriter

    def list_of_scalars_summary(self, tag_value_pairs, step):#记录scalar variables
        """Log scalar variables."""
        summary = tf.Summary(value=[tf.Summary.Value(tag=tag, simple_value=value) for tag, value in tag_value_pairs])
        self.writer.add_summary(summary, step)
        #由于版本问题，tf.summary.FileWriter可能会报错，改为tf.compat.v1.summary.FileWriter

来源：CSDN

作者：weixin_37718439

链接：https://blog.csdn.net/weixin_37718439/article/details/104225845