繁体   English   中英

PYTHON 将数组写入文件TXT

[英]PYTHON write an array to a file TXT

我想为深度学习数据创建一个小文件,所以我创建了一个具有不同像素值的文件 RED、GREEN 和 BLUE,并为每个数组添加了 output class 的不同值,所以我使用了该代码:

img=Image.open('Image.bmp')
M = img.crop((0,50,20,70))
M.show()
cv2.waitKey(0)
for j in range(20):
    for i in range(20):
        [r,v,b] = M.getpixel((i,j))
        f=open('C:\image echantillon\DATATR', 'a')
        f.write(str([r,v,b,1,0,0,0,0,0,0,0,0])+"\n" )
        f.close() 

我重复该代码 9 次,因为我的图像中有 9 个类,并且在我想加载该文件之后,但我想分割每一行以输入数据和 output 数据,所以我使用这个代码

data = np.genfromtxt('C:\image echantillon\DATATe',delimiter =',', dtype=str)
x=data[:,0:3] 
y=data[:,3:]

但是我以这种形式获得了数据,因此在进行深度学习之后我无法使用它

['[4' ' 3' ' 5'] [' 1' ' 0' ' 0' ' 0' ' 0' ' 0' ' 0' ' 0' ' 0]']
['[4' ' 3' ' 5'] [' 1' ' 0' ' 0' ' 0' ' 0' ' 0' ' 0' ' 0' ' 0]']
['[4' ' 3' ' 5'] [' 1' ' 0' ' 0' ' 0' ' 0' ' 0' ' 0' ' 0' ' 0]']

正确的形式是这样的:

[[' 4' ' 3' ' 5'] [' 1' ' 0' ' 0' ' 0' ' 0' ' 0' ' 0' ' 0' ' 0']]

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM