Limit on the keypoint dataset

tengteng95 / Pose-Transfer

Code for the paper Progressive Pose Attention for Person Image Generation in CVPR19 (Oral).

MIT License

734 stars 160 forks source link

Limit on the keypoint dataset #52

Closed cgokce closed 4 years ago

cgokce commented 4 years ago

Firstly, thanks for the great work and open source code.

At the data/keypoint.py file, you have manually set the dataset length as 4000. It limits training set to 4000 image pairs, and I believe this should be changed to return self.size.

    def __len__(self):
        if self.opt.phase == 'train':
            return 4000

Maybe I'm missing something. Is there a reason for limiting the dataset?

cgokce commented 4 years ago

Nevermind, I've found the answer(https://github.com/tengteng95/Pose-Transfer/issues/17).