|
|
|
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) |
|
|
|
|