svip-lab / impersonator

PyTorch implementation of our ICCV 2019 paper: Liquid Warping GAN: A Unified Framework for Human Motion Imitation, Appearance Transfer and Novel View Synthesis
https://svip-lab.github.io/project/impersonator
Other
1.73k stars 318 forks source link

Outputs results folder all empty #44

Closed smalleight17 closed 4 years ago

smalleight17 commented 4 years ago

Hi, I'm trying to run demo_imitator and everything seems to be running fine except all the folder under outputs/results/demos/imitators is empty. Any idea what I can do to fix this? Much appreciated! Here's the attached output from terminal:

------------ Options -------------
T_pose: False
batch_size: 1
bg_ks: 13
bg_model: ./outputs/checkpoints/deepfillv2/net_epoch_50_id_G.pth
bg_replace: False
body_seg: False
cam_strategy: smooth
checkpoints_dir: ./outputs/checkpoints/
cond_nc: 3
data_dir: /p300/datasets/iPER
dataset_mode: iPER
debug: False
do_saturate_mask: False
face_model: assets/pretrains/sphere20a_20171020.pth
front_warp: False
ft_ks: 3
gen_name: impersonator
gpu_ids: 0
has_detector: False
hmr_model: assets/pretrains/hmr_tf2pt.pth
image_size: 256
images_folder: images_HD
ip: 
is_train: False
load_epoch: 0
load_path: ./outputs/checkpoints/lwb_imper_fashion_place/net_epoch_30_id_G.pth
map_name: uv_seg
model: impersonator
n_threads_test: 2
name: running
norm_type: instance
only_vis: False
output_dir: ./outputs/results/
part_info: assets/pretrains/smpl_part_info.json
port: 31100
post_tune: False
pri_path: ./assets/samples/A_priors/imgs
repeat_num: 6
save_res: False
serial_batches: False
smpl_model: assets/pretrains/smpl_model.pkl
smpls_folder: smpls
src_path: 
swap_part: body
test_ids_file: val.txt
tex_size: 3
tgt_path: 
time_step: 10
train_ids_file: train.txt
uv_mapping: assets/pretrains/mapper.txt
view_params: R=0,90,0/t=0,0,0
-------------- End ----------------
./outputs/checkpoints/running
  0%|                                                                                                                                                    | 0/3 [00:00<?, ?it/s]Network impersonator was created
Loading net: ./outputs/checkpoints/lwb_imper_fashion_place/net_epoch_30_id_G.pth
Network deepfillv2 was created
Loading net: ./outputs/checkpoints/deepfillv2/net_epoch_50_id_G.pth

            Personalization: meta imitation...

100%|███████████████████Personalization: meta cycle finetune...████████████████████████████████████████████████████████████████████████████████| 25/25 [00:02<00:00,  9.58it/s]
load face model from assets/pretrains/sphere20a_20171020.pth
./outputs/results/demos/imitators/mixamo_preds
100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5/5 [00:49<00:00,  9.85s/it./outputs/results/demos/imitators/mixamo_preds███████████████████████████████████████████████████████████████████████████████████████████████| 148/148 [00:00<00:00, 830.61it/s]
                                                                                                                                                                              ./outputs/results/demos/imitators/mixamo_preds███████████████████████████████████████████████████████████████████████████████████████████████| 576/576 [00:00<00:00, 842.92it/s]
 33%|██████████████████████████████████████████████▎                                                                                            | 1/3 [01:46<03:32, 106.16s/it]Network impersonator was created████████████████████████████████████████████████████████████████████████████████████████████████████████████| 111/111 [00:00<00:00, 745.93it/s]
Loading net: ./outputs/checkpoints/lwb_imper_fashion_place/net_epoch_30_id_G.pth
Network deepfillv2 was created
Loading net: ./outputs/checkpoints/deepfillv2/net_epoch_50_id_G.pth

            Personalization: meta imitation...

100%|███████████████████Personalization: meta cycle finetune...████████████████████████████████████████████████████████████████████████████████| 25/25 [00:02<00:00, 10.11it/s]
load face model from assets/pretrains/sphere20a_20171020.pth
./outputs/results/demos/imitators/mixamo_preds
100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5/5 [00:48<00:00,  9.80s/it./outputs/results/demos/imitators/mixamo_preds███████████████████████████████████████████████████████████████████████████████████████████████| 148/148 [00:00<00:00, 793.41it/s]
                                                                                                                                                                              ./outputs/results/demos/imitators/mixamo_preds███████████████████████████████████████████████████████████████████████████████████████████████| 576/576 [00:00<00:00, 891.59it/s]
 67%|████████████████████████████████████████████████████████████████████████████████████████████▋                                              | 2/3 [03:27<01:44, 104.62s/it]Network impersonator was created████████████████████████████████████████████████████████████████████████████████████████████████████████████| 111/111 [00:00<00:00, 726.58it/s]
Loading net: ./outputs/checkpoints/lwb_imper_fashion_place/net_epoch_30_id_G.pth
Network deepfillv2 was created
Loading net: ./outputs/checkpoints/deepfillv2/net_epoch_50_id_G.pth

            Personalization: meta imitation...

100%|███████████████████Personalization: meta cycle finetune...████████████████████████████████████████████████████████████████████████████████| 25/25 [00:02<00:00, 10.18it/s]
load face model from assets/pretrains/sphere20a_20171020.pth
./outputs/results/demos/imitators/mixamo_preds
100%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 5/5 [00:48<00:00,  9.76s/it./outputs/results/demos/imitators/mixamo_preds███████████████████████████████████████████████████████████████████████████████████████████████| 148/148 [00:00<00:00, 720.35it/s]
                                                                                                                                                                              ./outputs/results/demos/imitators/mixamo_preds███████████████████████████████████████████████████████████████████████████████████████████████| 576/576 [00:00<00:00, 824.25it/s]
100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 3/3 [05:06<00:00, 103.17s/it]
Completed! All demo videos are save in ./outputs/results/demos/imitators████████████████████████████████████████████████████████████████████| 111/111 [00:00<00:00, 809.94it/s]
piaozhx commented 4 years ago

Have you ever installed ffmpeg?

smalleight17 commented 4 years ago

Works after I installed ffmpeg! Sorry my bad for missing that line of requirements! One question, --gpu_ids 0 works for me instead of --gpu_ids 1 (I have only one GPU). When I do --gpu_ids 1 I got error saying "torch.cuda.is_available() is False". Any reason for using --gpu_ids 1?

piaozhx commented 4 years ago

No special reason, you can set gpu_ids to any number you want as long as you have corresponding gpu.