关于openCV里imshow()显示图片的问题

xjtu_wong

1. import cv2
   
2. import os
   
3. import os.path
   
4. import numpy
   
5. 
   
6. def img2txt(img_path = r'C:\Users\LuvPeppa\Desktop\img_files\1',txt_name = 'new_test.txt'):
   
7.     R = []
   
8.     G = []
   
9.     B = []
   
10.     with open(txt_name,'w') as f:
    
11.         for each_file in os.listdir(img_path):
    
12.             #path = r'C:\Users\Wongtao\Desktop\Cifarpy\img_files\1'
    
13.             if os.path.splitext(each_file)[1] == '.jpg':
    
14.                 image = cv2.imread(each_file)        
    
15.                 p1=cv2.resize(image,(32,32),interpolation=cv2.INTER_CUBIC)
    
16.                 #each_data = numpy.reshape(p1,(32,32,3))
    
17.                 print('数组p1的形状是',numpy.shape(p1),type(p1))
    
18.                 for each_pix in p1:
    
19.                     for each_color in each_pix:
    
20.                         B.append(each_color[0])
    
21.                         G.append(each_color[1])
    
22.                         R.append(each_color[2])
    
23.                 new_data = list(R+G+B)
    
24.                 print('数组R的形状是%s'%numpy.shape(R))
    
25.                 print('数组new_data的形状是%s'%numpy.shape(new_data))
    
26.                 #for each_line in new_data:
    
27.                 new_data = str(new_data).replace(',',' ')
    
28.                 new_data = new_data.replace('[','')
    
29.                 new_data = new_data.replace(']','')
    
30.                 f.write(str(new_data)+'\n')
    
31.                 R = []
    
32.                 G = []
    
33.                 B = []
    
34.                 cv2.imshow('a',p1)
    
35.                 #cv2.imwrite('new_'+each_file,p1)#保存图片
    
36.                 print(p1)
    
37.                 break
    
38. 
    
39. def txt2img(txt_name = 'new_test.txt'):
    
40.     with open(txt_name,'r') as f:
    
41.         img_data = [];R = [];G = [];B =[];col = ['[',']']
    
42.         img_mat = numpy.zeros((1024,3))
    
43.         for each_line in f:
    
44.             img_data = [];R = [];G = [];B =[];
    
45.             print(type(each_line),len(each_line),'1')
    
46.             #print(each_line)   
    
47.             print(type(each_line),len(each_line),'2')
    
48.             each_line,nothing = each_line.split(sep = '\n')
    
49.             each_line = each_line.split(sep = '  ')
    
50.             print(type(each_line),len(each_line))
    
51.             for each_color in each_line:
    
52.                 #if each_color not in col:
    
53.                 img_data.append(int(each_color))
    
54.                 R = numpy.array(img_data[0:1024])
    
55.                 G = numpy.array(img_data[1024:2048])
    
56.                 B = numpy.array(img_data[2048:3072])
    
57.             #img_mat = numpy.array(numpy.hstack((B,G,R)))
    
58.             for each in range(len(B)):
    
59.                 img_mat[each][0] = int(B[each])
    
60.                 img_mat[each][1] = int(G[each])
    
61.                 img_mat[each][2] = int(R[each])
    
62.             #img_mat = img_mat.T
    
63.             #print('B',type(B))
    
64.             print(B[0],G[0],R[0])
    
65.             print('%s的形状是%s,类型是%s'%('img_mat',numpy.shape(img_mat),type(img_mat)),img_mat[0])
    
66.             cv2.imshow('a',numpy.reshape(img_mat,(32,32,3)))
    
67.             print(img_mat)
    
68.             break
    
69.    
    
70. if __name__ == '__main__':
    
71.     img2txt()
    
72.     txt2img()
    
73. 
    

复制代码

这段代码段img2txt()可以将图片转换为txt文档，但是txt2img()没法将该文档显示成图片。
明明打印出来的图片数据都一样，请问这是什么原因呢？

xjtu_wong

已解决

1. import cv2
   
2. import os
   
3. import os.path
   
4. import numpy
   
5. 
   
6. def img2txt(img_path = r'C:\Users\LuvPeppa\Desktop\img_files\1',txt_name = 'new_test.txt'):
   
7.     R = [];G = [];B = [];extentions = ['.jpg','.bmp','.png']
   
8.     with open(txt_name,'w') as f:
   
9.         type_no = 0
   
10.         for each_file in os.listdir(img_path):
    
11.             #path = r'C:\Users\Wongtao\Desktop\Cifarpy\img_files\1'
    
12.             type_no = type_no+1
    
13.             if os.path.splitext(each_file)[1] in extentions:
    
14.                 image = cv2.imread(each_file)        
    
15.                 p1=cv2.resize(image,(32,32))
    
16.                 for each_pix in p1:
    
17.                     for each_color in each_pix:
    
18.                         B.append(each_color[0])
    
19.                         G.append(each_color[1])
    
20.                         R.append(each_color[2])
    
21.                 new_data = list(R+G+B)
    
22.                 new_data = str(new_data).replace(',',' ')
    
23.                 new_data = new_data.replace('[','')
    
24.                 new_data = new_data.replace(']','')
    
25.                 f.write(str(new_data)+'\n')
    
26.                 R = []
    
27.                 G = []
    
28.                 B = []
    
29.                 #cv2.imshow('a',p1)
    
30.                 #os.chdir(img_path+'\\'+str(type_no))
    
31.                 print('图像%s正在保存。。。'%('new_'+each_file))
    
32.                 #cv2.imwrite('new_'+each_file,p1)#保存图片
    
33.                 #os.chdir(img_path)
    
34.                
    
35. 
    
36. def txt2img(txt_name = 'new_test.txt'):
    
37.     with open(txt_name,'r') as f:
    
38.         pic_num = 0
    
39.         for each_line in f:
    
40.             pic_num = pic_num+1
    
41.             img_data = [];R = [];G = [];B =[];img_mat = []
    
42.             each_line,nothing = each_line.split(sep = '\n')
    
43.             each_line = each_line.split(sep = '  ')
    
44.             for each_color in each_line:
    
45.                 img_data.append(int(each_color))
    
46.                 R = bytearray(img_data[0:1024])
    
47.                 G = bytearray(img_data[1024:2048])
    
48.                 B = bytearray(img_data[2048:3072])
    
49.             for index in range(1024):
    
50.                 img_mat.append(B[index])
    
51.                 img_mat.append(G[index])
    
52.                 img_mat.append(R[index])
    
53.             img_mat = bytearray(img_mat)
    
54.             img_mat = numpy.array(img_mat)
    
55.             bgrImage = img_mat.reshape(32,32,3)
    
56.             print('图像%s正在显示。。。'%pic_num)
    
57.             #cv2.imshow(str(pic_num)+'_back.png', bgrImage)
    
58.             # print('图像%s正在保存。。。'%pic_num)
    
59.             #cv2.imwrite(str(pic_num)+'_back.png', bgrImage)
    
60.             
    
61.             
    
62. if __name__ == '__main__':
    
63.     img2txt()
    
64.     txt2img()
    
65. 
    

复制代码