ZhihengCV / Bayesian-Crowd-Counting

Official Implement of ICCV 2019 oral paper Bayesian Loss for Crowd Count Estimation with Point Supervision
330 stars 84 forks source link

What is the shape of the .npy file? #47

Open xiaotongtongxue opened 5 months ago

xiaotongtongxue commented 5 months ago

Hi ! When I use my own dataset, I have a problem about the keypoints, which is File "D:\Crowed_recognition\code\GeneralizedLoss-Counting-Pytorch-main\datasets\crowd.py", line 112, in train_transform_with_crop nearest_dis = np.clip(keypoints[:, 2], 4.0, 128.0) IndexError: index 2 is out of bounds for axis 1 with size 2.

By the way, the shape of my own .npy file is (n, 2), such as [[1,2],[2,3]]. I do not know the real shape of the .npy file. Pls help, thanks!