版权声明:转载请注明出处 https://blog.csdn.net/sinat_39372048/article/details/82784859
#coding: utf-8
import h5py
import matplotlib.pyplot as plt
from skimage import io,color
import numpy as np
for j in range(9):
imgt = io.imread('t{0}.BMP'.format(j+1))
imgl = io.imread('l{0}.BMP'.format(j+1))
datatr1 = color.rgb2gray(imgt)
datala1 = color.rgb2gray(imgl)
datatr2 = color.rgb2gray(imgt)
datala2 = color.rgb2gray(imgl)
datatr3 = color.rgb2gray(imgt)
datala3 = color.rgb2gray(imgl)
#shangxia
for i in range(60):
datatr1[[i*8, i*8+7], :] = datatr1[[i*8+7, i*8], :]
datatr1[[i*8+1, i*8+6], :] = datatr1[[i*8+6, i*8+1], :]
datatr1[[i*8+2, i*8+5], :] = datatr1[[i*8+5, i*8+2], :]
datatr1[[i*8+3, i*8+4], :] = datatr1[[i*8+4, i*8+3], :]
datala1[[i*8, i*8+7], :] = datala1[[i*8+7, i*8], :]
datala1[[i*8+1, i*8+6], :] = datala1[[i*8+6, i*8+1], :]
datala1[[i*8+2, i*8+5], :] = datala1[[i*8+5, i*8+2], :]
datala1[[i*8+3, i*8+4], :] = datala1[[i*8+4, i*8+3], :]
with h5py.File('h6{0}.h5'.format(j),'w') as f:
f.create_dataset('trainimg',data=datatr1)
f.create_dataset('labelimg',data=datala1)
#zuoyou
for i in range(104):
datatr2[:,[i*8, i*8+7]] = datatr2[:,[i*8+7, i*8]]
datatr2[:,[i*8+1, i*8+6]] = datatr2[:,[i*8+6, i*8+1]]
datatr2[:,[i*8+2, i*8+5]] = datatr2[:,[i*8+5, i*8+2]]
datatr2[:,[i*8+3, i*8+4]] = datatr2[:,[i*8+4, i*8+3]]
datala2[:,[i*8, i*8+7]] = datala2[:,[i*8+7, i*8]]
datala2[:,[i*8+1, i*8+6]] = datala2[:,[i*8+6, i*8+1]]
datala2[:,[i*8+2, i*8+5]] = datala2[:,[i*8+5, i*8+2]]
datala2[:,[i*8+3, i*8+4]] = datala2[:,[i*8+4, i*8+3]]
with h5py.File('h7{0}.h5'.format(j),'w') as f:
f.create_dataset('trainimg',data=datatr2)
f.create_dataset('labelimg',data=datala2)
#zhuanzhi
datatr3 = np.transpose(datatr3)
datala3 = np.transpose(datala3)
with h5py.File('h8{0}.h5'.format(j),'w') as f:
f.create_dataset('trainimg',data=datatr3)
f.create_dataset('labelimg',data=datala3)
#shangzhuan
datatr1 = np.transpose(datatr1)
datala1 = np.transpose(datala1)
with h5py.File('h9{0}.h5'.format(j),'w') as f:
f.create_dataset('trainimg',data=datatr1)
f.create_dataset('labelimg',data=datala1)
#zuozhuan
datatr2 = np.transpose(datatr2)
datala2 = np.transpose(datala2)
with h5py.File('h10{0}.h5'.format(j),'w') as f:
f.create_dataset('trainimg',data=datatr2)
f.create_dataset('labelimg',data=datala2)