caffe 绘制acceracy曲线 IndexError: list index out of range的解决方案

来源:互联网 发布:大数据修炼系统无错 编辑:程序博客网 时间:2024/06/06 07:10

在绘制caffe的loss曲线时报错:

解决方案:首先考虑权限问题,sudo运行即可,如果解决不了则修改。/tools/extra/plot_training_log.py

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:            fields = line.split()if len(fields) != num_fields:        continue                data[0].append(float(fields[field_idx0].strip()))                data[1].append(float(fields[field_idx1].strip()))    return data


阅读全文
0 1