Incorrect references to validation and training sets

wbenbihi / hourglasstensorflow

Tensorflow implementation of Stacked Hourglass Networks for Human Pose Estimation

MIT License

480 stars 179 forks source link

Incorrect references to validation and training sets #61

Closed xiaoxin05 closed 2 years ago

xiaoxin05 commented 4 years ago

Hi, thank you for sharing your work。Finding problems in the learning process, in hourglass_tiny.py ,It's about line 311（You may not be able to match the number of lines because I added some comments yourself） img_valid, gt_valid, w_valid = next(self.generator)，I think is img_valid, gt_valid, w_valid = next(self.valid_gen)