search

michalfaber / tensorflow_Realtime_Multi-Person_Pose_Estimation

Multi-Person Pose Estimation project for Tensorflow 2.0 with a small and fast model based on MobilenetV3
Other
215 stars 65 forks source link

example.py shows error when mobilenet heatmap.npy and paf.npy is loaded #7

Open suginesh9746 opened 4 years ago

suginesh9746 commented 4 years ago

when i try example.py with heatmap and paf generated by mobilenet architecture , the following error was thrown

IndexError Traceback (most recent call last)

in 22 #print(coordinates) 23 ---> 24 connections = get_connections(cfg, coordinates, paf) 25 #print(connections) 26 skeletons = estimate(cfg, connections) in get_connections(config, coords, paf, threshold, mid_num, minimum_mid_num) 28 29 # select dx and dy PAFs for this connection type ---> 30 paf_dx = paf[:, :, conn.paf_dx_idx] 31 #print(conn.paf_dx_idx) 32 paf_dy = paf[:, :, conn.paf_dy_idx] IndexError: index 28 is out of bounds for axis 2 with size 28 [example.zip](https://github.com/michalfaber/tensorflow_Realtime_Multi-Person_Pose_Estimation/files/4253483/example.zip) iam attaching a zip file having example.ipynb which is my code and also it contains heatmap.npy and paf.npy