search

AliaksandrSiarohin / pose-gan

381 stars 78 forks source link

demo.py ValueError #54

Open jihwan1008 opened 4 years ago

jihwan1008 commented 4 years ago

Hi,

I'm trying to run demo.py, whereas I keep facing ValueError: You are trying to load a weight file containing 25 layers into a model with 21 layers.. Since there are variety of pre-trained models, I'm not sure how to figure out the proper pre-trained model for demo.py. I printed out len(model.layers), but it spits out the wrong number.

Any help would be deeply appreciated. Thanks in advance.

AliaksandrSiarohin commented 4 years ago

Hi, the fashion checkpoints have more layers than market. Plus the number of layers depend on the model with dsc or without it, please refer to https://github.com/AliaksandrSiarohin/pose-gan/issues/47 . So in demo.py you should specify is your model use dsc and if the model is trained on market or fashion.