faceparsing traning issue

switchablenorms / CelebAMask-HQ

A large-scale face dataset for face parsing, recognition, generation and editing.

2.04k stars 343 forks source link

faceparsing traning issue #97

Open leexuan0128 opened 2 years ago

leexuan0128 commented 2 years ago

Line 135 in trainer.py, generate_label( ) miss the img_size parameter?

stove92 commented 2 years ago

It suppose to use self.imsize as the second parameter, just like line 114 and line 115 in trainer.py. Also line 136 should be deleted because torch.from_numpy has beed called inside the generate_label funcion. Am I right?