NVlabs / imaginaire

NVIDIA's Deep Imagination Team's PyTorch Library
Other
3.99k stars 444 forks source link

[street] The image synthesized during training is black #153

Open songyn95 opened 2 years ago

songyn95 commented 2 years ago

https://github.com/NVlabs/few-shot-vid2vid/

I am going to synthesize traffic video(2 sequences in the training set, 160 images for each). During the training, I found that the synthetic image was black during iteration 14/70. What is the reason and how to solve it? The training and test commands are as follows:

  1. python train.py --name test --dataset_mode fewshot_street --adaptive_spade --loadSize 512 --fineSize 512 --batchSize 2 --dataroot datasets --label_nc 0 --gpu_ids 2
  2. python test.py --name test --dataset_mode fewshot_street --adaptive_spade --loadSize 512 --fineSize 512 --seq_path datasets/A_01 --ref_img_path datasets/B_01 --label_nc 0 --gpu_ids 3
YukunZEX commented 6 months ago

i got same problem in old version of fsvid2vid project, have you solve it?