图像预处理,旋转,翻转,转置等

版权声明:转载请注明出处 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)



猜你喜欢

转载自blog.csdn.net/sinat_39372048/article/details/82784859
今日推荐