microsoft / multiview-human-pose-estimation-pytorch

This is an official Pytorch implementation of "Cross View Fusion for 3D Human Pose Estimation, ICCV 2019".
MIT License
541 stars 89 forks source link

operands could not be broadcast together with shapes (0,) (0,17,2) #23

Closed Felix1014 closed 4 years ago

Felix1014 commented 4 years ago

Hi, first thanks for your work. When I run the 2d train code, it reminds me the following errors. I don't know how to solve this. Could you help me find the problems?

Epoch: [0][0/696] Time 5.216s (5.216s) Speed 6.1 samples/s Data 2.286s (2.286s) Loss 0.36521 (0.36521) Accuracy 0.026 (0.026) Memory 1429799424.0 Epoch: [0][100/696] Time 0.424s (0.543s) Speed 75.4 samples/s Data 0.000s (0.041s) Loss 0.34272 (0.39108) Accuracy 0.000 (0.002) Memory 1429799424.0 Epoch: [0][200/696] Time 0.453s (0.518s) Speed 70.7 samples/s Data 0.000s (0.028s) Loss 0.37073 (0.37585) Accuracy 0.015 (0.002) Memory 1428750848.0 Epoch: [0][300/696] Time 0.495s (0.506s) Speed 64.7 samples/s Data 0.000s (0.023s) Loss 0.34146 (0.36804) Accuracy 0.133 (0.016) Memory 1429799424.0 Epoch: [0][400/696] Time 0.503s (0.500s) Speed 63.6 samples/s Data 0.000s (0.020s) Loss 0.26815 (0.35295) Accuracy 0.424 (0.076) Memory 1428750848.0 Epoch: [0][500/696] Time 0.488s (0.495s) Speed 65.5 samples/s Data 0.000s (0.018s) Loss 0.25259 (0.33488) Accuracy 0.460 (0.149) Memory 1429799424.0 Epoch: [0][600/696] Time 0.423s (0.492s) Speed 75.7 samples/s Data 0.000s (0.017s) Loss 0.23425 (0.31868) Accuracy 0.551 (0.212) Memory 1428750848.0 Traceback (most recent call last): File "run/pose2d/train.py", line 189, in main() File "run/pose2d/train.py", line 164, in main criterion, final_output_dir, writer_dict) File "/lustre/alice3/scratch/3dpoint/fz64/project/multiview/run/pose2d/../../lib/core/function.py", line 233, in validate name_value, perf_indicator = dataset.evaluate(all_preds) File "/lustre/alice3/scratch/3dpoint/fz64/project/multiview/run/pose2d/../../lib/dataset/multiview_h36m.py", line 145, in evaluate distance = np.sqrt(np.sum((gt - pred)**2, axis=2)) ValueError: operands could not be broadcast together with shapes (0,) (0,17,2)