def transfer(self):
image_reshape = np.ndarray(shape=(self.pre_images.shape[0], self.output_rows, self.output_cols, 3),
dtype=np.float16)
widgets = ['Transfer: ', pbar.Percentage(), ' ', pbar.Bar('>'), ' ', pbar.ETA()]
image_bar = pbar.ProgressBar(widgets=widgets, maxval=self.pre_images.shape[0]).start()
for i in range(0, self.pre_images.shape[0]):
image = self.pre_images[i].reshape(self.pre_img_rows, self.pre_img_cols)
image = image.astype('uint8')
im = Image.fromarray(image) # monochromatic image
imrgb = im.convert('RGB')
imrgb = imrgb.resize((self.output_rows, self.output_cols), Image.ANTIALIAS)
im = np.array(imrgb, dtype=np.float16)
im[:, :, 0] -= imagenet_mean['R']
im[:, :, 1] -= imagenet_mean['G']
im[:, :, 2] -= imagenet_mean['B']
# 'RGB'->'BGR', historical reasons in OpenCV
im = im[:, :, ::-1]
image_reshape[i] = im
# test for correct convert!
# if i < 3:
# img = Image.fromarray(np.uint8(im))
# img.save(str(i) + '.jpeg', 'jpeg')
image_bar.update(i + 1)
image_bar.finish()
print('image_reshape:', image_reshape.shape)
return image_reshape
评论列表
文章目录