search

yanx27 / EverybodyDanceNow_reproduce_pytorch

Everybody dance now reproduced in pytorch
MIT License
605 stars 174 forks source link

About the config of pix2pixHD which didn't use the local generator #83

Closed WilliammmZ closed 4 years ago

WilliammmZ commented 4 years ago

The config of the pix2pixHD at train.py, the param 'netG' set to 'global'. In my opinion, it causes that the model only uses the global generator which more like pix2pix not pix2pixHD. I consider the param should be changed to 'local' for use both global and local generator~

ZJ9527 commented 4 years ago

hello,i met a qusetion when i run normalization.py,i try my best to solve it,but i failed,if you have some advice? the following is my error File "normalization.py", line 74, in new_sourcehead, = get_scale(new_source[:,:,0]) File "normalization.py", line 43, in get_scale head_x, height = linspace1[list(any1)][0], len(linspace1[list(any1)]) IndexError: index 0 is out of bounds for axis 0 with size 0

WilliammmZ commented 4 years ago

In my mind, the line43's code try to find the height pixel of the image, calculate the height of the dancer and confirm the position of the head. I think the problem may caused by the code in line70, please check the data in 'new_source'. Thanks~