保存list数据:
file = open('feature.txt', 'w')
file.write(str(list_data))
file.close()
with open(feature_txt, 'w') as fp:
feature = net(X)
feature = feature.data.cpu().numpy()
fp.write(str(feature) + '\n') # dim [1, 678]
kk = 0
这样保存 numpy数据到txt文件,会有一个问题,见图片:
它不是绝对的一行一行的,这在试图读出来的时候,很不方便。
解决方案
To be continue