txt文本形式
{"tp": 0.0, "fp": 0.0, "fn": 2639.1206896551726, "tn": 284840.7586206897, "total_size": 287479.8793103448, "iou": 0.0, "precision": 0.0, "recall": 0.0, "f1": 0.0}
def txt_excel(filename,sheet):
with open(filename,'r') as f:
datas = f.readlines()
wb= load_workbook("score.xlsx")
sheet = wb.create_sheet(title="{}".format(sheet))
epoch = 1
sheet['A1'] = "Epoch"
sheet['B1'] = 'TP'
sheet['C1'] = 'FP'
sheet['D1'] = 'TN'
sheet['E1'] = "Total_size"
sheet['F1'] = 'IOU'
sheet['G1'] = 'Precision'
sheet['H1'] = 'Recall'
sheet['I1'] = 'F1'
for i in range(len(datas)):
data = datas[i]
data = json.loads(data)
row = str(i + 2)
sheet["A" + row] = epoch
epoch = epoch + 1
sheet["B" + row] = data['tp']
sheet["C" + row] = data['fp']
sheet["D" + row] = data['tn']
sheet["E" + row] = data['total_size']
sheet["F" + row] = data['iou']
sheet["G" + row] = data['precision']
sheet["H" + row] = data['recall']
sheet["I" + row] = data['f1']
wb.save('score.xlsx')```
import os
listname = os.listdir('./')
for name in listname:
if name.endswith('.txt'):
txt_excel(name,name.split('.')[0])