nikeaddidaslining

Results 2 issues of nikeaddidaslining

there is some problems in get_heatmap function

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

test CUDA out of memory

Why do I have this problem when I test： OutOfMemoryError: CUDA out of memory. Tried to allocate 1.71 GiB (GPU 0; 6.00 GiB total capacity; 3.49 GiB already allocated; 1.04...