-
Notifications
You must be signed in to change notification settings - Fork 23
Description
(sparnet) C:\Users\user\sparnet>python test_enhance_single_unalign.py --gpus 1 --model sparnethd --name SPARNetHD_V4_Attn2D --res_depth 10 --att_name spar --Gnorm "in" --pretrain_model_path pretrain_models/SPARNetHD_V4_Attn2D_net_H-epoch10.pth --test_img_path ./test_images/test_hzgg.jpg --results_dir test_hzgg_results
----------------- Options ---------------
Dnorm: none
Gnorm: in [default: bn]
att_name: spar
batch_size: 32
bottleneck_size: 4
checkpoints_dir: ./check_points
data_device: cuda:0 [default: None]
dataroot: None
dataset_name: celeba
debug: False
device: cuda:0 [default: None]
epoch: latest
eval: False
gpu_ids: [0] [default: None]
gpus: 1
init_gain: 0.02
init_type: normal
input_nc: 3
isTrain: False [default: None]
load_iter: 0 [default: 0]
load_size: None
max_dataset_size: inf
model: sparnethd [default: test]
n_layers_D: 4
name: SPARNetHD_V4_Attn2D [default: experiment_name]
ndf: 64
ngf: 64
ntest: inf
num_D: 3
num_test: 50
num_threads: 8
output_nc: 3
phase: test
preprocess: none
pretrain_model_path: pretrain_models/SPARNetHD_V4_Attn2D_net_H-epoch10.pth [default: ]
res_depth: 10
results_dir: test_hzgg_results [default: ./results/]
save_as_dir:
seed: 123
serial_batches: False
suffix:
test_img_path: ./test_images/test_hzgg.jpg [default: ]
test_upscale: 1
verbose: False
----------------- End -------------------
======> Loading images, crop and align faces.
======> Saving aligned LQ faces to test_hzgg_results\LQ_faces
initialize network SPARNet with normal
model [SPARNetHDModel] was created
Loading pretrained model pretrain_models/SPARNetHD_V4_Attn2D_net_H-epoch10.pth
100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 1/1 [08:16<00:00, 496.16s/it]
======> Save the enhanced faces.
======> Paste the enhanced faces back to the original image.
100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 1/1 [00:00<00:00, 8.58it/s]
======> Save final result to test_hzgg_results\hq_final.jpg
(sparnet) C:\Users\user\sparnet>
