caffe绘制loss和accuracy曲线

最新推荐文章于 2023-06-03 14:28:25 发布

原创最新推荐文章于 2023-06-03 14:28:25 发布 · 279 阅读

0 ·

CC 4.0 BY-SA版权

博客主要讲述了caffe绘制曲线时出现的问题，具体为提示数组越界。针对该问题给出了解决办法，即修改文件中的一个函数，并给出了修改后的内容。

caffe绘制曲线时出现问题

1.提示数组越界的问题

绘制曲线时出现下面的问题：

Traceback (most recent call last):
  File "./plot_training_log.py.example", line 201, in <module>
    plot_chart(chart_type, path_to_png, path_to_logs)
  File "./plot_training_log.py.example", line 134, in plot_chart
    data = load_data(data_file, x, y)
  File "./plot_training_log.py.example", line 88, in load_data
    data[1].append(float(fields[field_idx1].strip()))
IndexError: list index out of range

解决办法是修改plot_training_log.py.example文件中的一个函数，如下：

def load_data(data_file, field_idx0, field_idx1):
    data = [[], []]
    with open(data_file, 'r') as f:
        for line in f:
            line = line.strip()
            if line[0] != '#':
                fields = line.split()
                data[0].append(float(fields[field_idx0].strip()))
                data[1].append(float(fields[field_idx1].strip()))
    return data

修改成下面的内容：

def load_data(data_file, field_idx0, field_idx1):
    data = [[], []]
    with open(data_file, 'r') as f:
        for line in f:
            line = line.strip()
            if line[0] == '#':
                num_fields = len(line.split())
            else:
                fileds = line.split()
                if len(fileds) != num_fields:
                    continue
                data[0].append(float(fileds[field_idx0].strip()))
                data[1].append(float(fileds[field_idx1].strip()))
    return data