在绘制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