File size: 1,202 Bytes
2d7efb8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
import os
import numpy as np
import argparse
from manipulate import Manipulator
from PIL import Image
#%%
if __name__ == "__main__":
parser = argparse.ArgumentParser(description='Process some integers.')
parser.add_argument('--dataset_name',type=str,default='ffhq',
help='name of dataset, for example, ffhq')
args = parser.parse_args()
dataset_name=args.dataset_name
if not os.path.isdir('./data/'+dataset_name):
os.system('mkdir ./data/'+dataset_name)
#%%
M=Manipulator(dataset_name=dataset_name)
np.set_printoptions(suppress=True)
print(M.dataset_name)
#%%
M.img_index=0
M.num_images=50
M.alpha=[0]
M.step=1
lindex,bname=0,0
M.manipulate_layers=[lindex]
codes,out=M.EditOneC(bname)
#%%
for i in range(len(out)):
img=out[i,0]
img=Image.fromarray(img)
img.save('./data/'+dataset_name+'/'+str(i)+'.jpg')
#%%
w=np.load('./npy/'+dataset_name+'/W.npy')
tmp=w[:M.num_images]
tmp=tmp[:,None,:]
tmp=np.tile(tmp,(1,M.Gs.components.synthesis.input_shape[1],1))
np.save('./data/'+dataset_name+'/w_plus.npy',tmp)
|