there is some problems in get_heatmap function

[nikeaddidaslining]

single_heatmap = np.zeros(shape=(heatmap_height, heatmap_width), dtype=np.float32) should in the loop: for j in range(keypoints.shape[0]):,