Face-SPARNet
Face-SPARNet copied to clipboard
Not working
(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>