python 獲取文件名+將損失寫進txt

獲取帶格式的和不帶文件名稱 

str='dataset/train/labels\\aaaaa.png'
s = str.split('\\')[-1]  # 選取後綴名爲 py 的文件  得到aaaaa.png
s = s.split('.')[0]  #得到aaaaa
print(s)

將訓練中的損失保存到txt文件中

#txt所在的路徑
loss_save='lossRecord/loss.txt'
#打開文件，mode='a'表示可以寫入多行，mode='w'我是使用後只能寫入1行
file_save=open(args.loss_save,mode='a')
#write前面有個換行，即先執行換行操作，然後將需要保存的值寫入
file_save.write('\n'+'step:'+str(step)+'  cross_entroy:'+str(loss_tr)+'  l2_loss:'+str(l2_loss_tr))
#關閉文件
file_save.close()