关于openCV里imshow()显示图片的问题
import cv2import 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) == '.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)
G.append(each_color)
R.append(each_color)
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)
G = numpy.array(img_data)
B = numpy.array(img_data)
#img_mat = numpy.array(numpy.hstack((B,G,R)))
for each in range(len(B)):
img_mat = int(B)
img_mat = int(G)
img_mat = int(R)
#img_mat = img_mat.T
#print('B',type(B))
print(B,G,R)
print('%s的形状是%s,类型是%s'%('img_mat',numpy.shape(img_mat),type(img_mat)),img_mat)
cv2.imshow('a',numpy.reshape(img_mat,(32,32,3)))
print(img_mat)
break
if __name__ == '__main__':
img2txt()
txt2img()
这段代码段img2txt()可以将图片转换为txt文档，但是txt2img()没法将该文档显示成图片。
明明打印出来的图片数据都一样，请问这是什么原因呢？ 已解决
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) 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)
G.append(each_color)
R.append(each_color)
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)
G = bytearray(img_data)
B = bytearray(img_data)
for index in range(1024):
img_mat.append(B)
img_mat.append(G)
img_mat.append(R)
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()
页:
[1]