Augmentorを用いて画像変換する際に、出力データを入力データと同じ形式で出力したい。

あるRGB画像データをAugmentorを用いて拡大し、新たなデータのサイズを確認しました。

(28,28)の784のデータが(3,28,28)の3倍のデータ量になりました。

ただ私の理想は784のデータを拡大するが、出力するデータ量は同じ(28,28)で出力してほしいと思っています。

そのように同じデータで画像変換することは可能でしょうか。

python
1import numpy as np
2import cv2
3from PIL import Image
4
5
6img = //入力データ画像
7img = img.reshape(28,28)
8
9# 画像として保存
10img = Image.fromarray(img)
11img.convert("RGB").save("../.png","PNG")
12
13# Augmentorを用いて画像を拡大
14p = Augmentor.Pipeline("../dl/picture/")
15#p.rotate(probability=0.7, max_left_rotation=10, max_right_rotation=10)
16p.zoom(probability=1.0, min_factor=1.5, max_factor=1.5)
17p.sample(1)
18
19img_ = // 変換した画像データ 
20Image.open("../.png")
21
22
23img_ = np.asarray(img_)
24plt.imshow(img_[:,:,1],cmap="gray")
25#img_ = img_.reshape(3,28,28)
26print(img_.shape)
27
28# →　結果を見るとデータが(1,28,28)→(3,28,28)となり、3倍のデータの量となっています。