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

xjtu_wong

import cv2
import os
import os.path
import numpy

def img2txt(img_path = r'C:\Users\LuvPeppa\Desktop\img_files\1',txt_name = 'new_test.txt'):
    R = []
    G = []
    B = []
    with open(txt_name,'w') as f:
        for each_file in os.listdir(img_path):
            #path = r'C:\Users\Wongtao\Desktop\Cifarpy\img_files\1'
            if os.path.splitext(each_file)[1] == '.jpg': 
                image = cv2.imread(each_file)        
                p1=cv2.resize(image,(32,32),interpolation=cv2.INTER_CUBIC)
                #each_data = numpy.reshape(p1,(32,32,3))
                print('数组p1的形状是',numpy.shape(p1),type(p1))
                for each_pix in p1:
                    for each_color in each_pix:
                        B.append(each_color[0])
                        G.append(each_color[1])
                        R.append(each_color[2])
                new_data = list(R+G+B)
                print('数组R的形状是%s'%numpy.shape(R))
                print('数组new_data的形状是%s'%numpy.shape(new_data))
                #for each_line in new_data:
                new_data = str(new_data).replace(',',' ')
                new_data = new_data.replace('[','')
                new_data = new_data.replace(']','')
                f.write(str(new_data)+'\n')
                R = []
                G = []
                B = []
                cv2.imshow('a',p1)
                #cv2.imwrite('new_'+each_file,p1)#保存图片
                print(p1)
                break

def txt2img(txt_name = 'new_test.txt'):
    with open(txt_name,'r') as f:
        img_data = [];R = [];G = [];B =[];col = ['[',']']
        img_mat = numpy.zeros((1024,3))
        for each_line in f:
            img_data = [];R = [];G = [];B =[];
            print(type(each_line),len(each_line),'1')
            #print(each_line)    
            print(type(each_line),len(each_line),'2')
            each_line,nothing = each_line.split(sep = '\n')
            each_line = each_line.split(sep = '  ')
            print(type(each_line),len(each_line))
            for each_color in each_line:
                #if each_color not in col:
                img_data.append(int(each_color))
                R = numpy.array(img_data[0:1024])
                G = numpy.array(img_data[1024:2048])
                B = numpy.array(img_data[2048:3072])
            #img_mat = numpy.array(numpy.hstack((B,G,R)))
            for each in range(len(B)):
                img_mat[each][0] = int(B[each])
                img_mat[each][1] = int(G[each])
                img_mat[each][2] = int(R[each])
            #img_mat = img_mat.T
            #print('B',type(B))
            print(B[0],G[0],R[0])
            print('%s的形状是%s,类型是%s'%('img_mat',numpy.shape(img_mat),type(img_mat)),img_mat[0])
            cv2.imshow('a',numpy.reshape(img_mat,(32,32,3)))
            print(img_mat)
            break
    
if __name__ == '__main__':
    img2txt()
    txt2img()

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

xjtu_wong

已解决

import cv2
import os
import os.path
import numpy

def img2txt(img_path = r'C:\Users\LuvPeppa\Desktop\img_files\1',txt_name = 'new_test.txt'):
    R = [];G = [];B = [];extentions = ['.jpg','.bmp','.png']
    with open(txt_name,'w') as f:
        type_no = 0
        for each_file in os.listdir(img_path):
            #path = r'C:\Users\Wongtao\Desktop\Cifarpy\img_files\1'
            type_no = type_no+1
            if os.path.splitext(each_file)[1] in extentions: 
                image = cv2.imread(each_file)        
                p1=cv2.resize(image,(32,32))
                for each_pix in p1:
                    for each_color in each_pix:
                        B.append(each_color[0])
                        G.append(each_color[1])
                        R.append(each_color[2])
                new_data = list(R+G+B)
                new_data = str(new_data).replace(',',' ')
                new_data = new_data.replace('[','')
                new_data = new_data.replace(']','')
                f.write(str(new_data)+'\n')
                R = []
                G = []
                B = []
                #cv2.imshow('a',p1)
                #os.chdir(img_path+'\\'+str(type_no))
                print('图像%s正在保存。。。'%('new_'+each_file))
                #cv2.imwrite('new_'+each_file,p1)#保存图片
                #os.chdir(img_path)
                

def txt2img(txt_name = 'new_test.txt'):
    with open(txt_name,'r') as f:
        pic_num = 0
        for each_line in f:
            pic_num = pic_num+1
            img_data = [];R = [];G = [];B =[];img_mat = []
            each_line,nothing = each_line.split(sep = '\n')
            each_line = each_line.split(sep = '  ')
            for each_color in each_line:
                img_data.append(int(each_color))
                R = bytearray(img_data[0:1024])
                G = bytearray(img_data[1024:2048])
                B = bytearray(img_data[2048:3072])
            for index in range(1024):
                img_mat.append(B[index])
                img_mat.append(G[index])
                img_mat.append(R[index])
            img_mat = bytearray(img_mat)
            img_mat = numpy.array(img_mat)
            bgrImage = img_mat.reshape(32,32,3)
            print('图像%s正在显示。。。'%pic_num)
            #cv2.imshow(str(pic_num)+'_back.png', bgrImage)
            # print('图像%s正在保存。。。'%pic_num)
            #cv2.imwrite(str(pic_num)+'_back.png', bgrImage)
            
            
if __name__ == '__main__':
    img2txt()
    txt2img()