search

zhiqiangdon / pose-adv-aug

Code for "Jointly Optimize Data Augmentation and Network Training: Adversarial Data Augmentation in Human Pose Estimation" (CVPR 2018)
Apache License 2.0
84 stars 14 forks source link

why is ZeroDivisionError: float division by zero when i train LSP? #6

Open zouxuelian opened 5 years ago

zouxuelian commented 5 years ago

Traceback (most recent call last): File "/media/pose-adv-aug/collect-scale-ditri.py", line 632, in main() File "/media/pose-adv-aug/collect-scale-ditri.py", line 393, in main train_distri_path_2, hg, opt, is_train=True) File "/media/pose-adv-aug/collect-scale-ditri.py", line 428, in collect_train_valid_data collect_data(collect_data_loader, dataset, hg, save_path) File "/media/pose-adv-aug/collect-scale-ditri.py", line 537, in collect_data grnd_pts, normalizer, rotation) File "/media/pose-adv-aug/pylib/Evaluation.py", line 366, in per_person_pckh dists = calc_dists(pred_pts, grnd_pts, normalizers, use_zero=True) File "/media/pose-adv-aug/pylib/Evaluation.py", line 291, in calc_dists dists[c, n] = torch.dist(preds[n, c, :], target[n, c, :]) / normalize[n] ZeroDivisionError: float division by zero