utils_combine.py 文件源码

def fetchdatalabel(path, postfix='roienhance.mat', flag='train'):  # 'enhance.mat' 'roienhance.jpeg'
  data = np.zeros((58, 40, 40))
  label = np.zeros((58, 40, 40))
  if flag == 'train':
    data = np.zeros((58*4, 40, 40))
    label = np.zeros((58*4, 40, 40))
  datacount = 0
  fname = []
  for file in os.listdir(path):
    if file.endswith(postfix):
      if postfix[-4:] == '.mat':
        im = sio.loadmat(path+file)
        im = im['im']
      elif postfix[-5:] == '.jpeg':
        im = scipy.misc.imread(path+file)
        im = im*1.0 / 255.0
      imlabel = sio.loadmat(path+file[:-len(postfix)]+'massgt.mat')
      imlabel = imlabel['im']
      data[datacount, :, :] = im
      label[datacount, :, :] = imlabel
      datacount += 1
      if flag == 'train':
        data[datacount, :, :] = im[:, ::-1]
        label[datacount, :, :] = imlabel[:, ::-1]
        data[datacount+1, :, :] = im[::-1, :]
        label[datacount+1, :, :] = imlabel[::-1, :]
        im1 = im[::-1, :]  # vertical flip, then horizontal flip
        imlabel1 = imlabel[::-1, :]
        data[datacount+2, :, :] = im1[:, ::-1]
        label[datacount+2, :, :] = imlabel1[:, ::-1] 
        datacount += 3
      fname.append(file)
  if flag == 'train': assert(datacount==58*4)
  else: assert(datacount==58)
  return data , label, fname