search

taohan10200 / IIM

PyTorch implementations of the paper: "Learning Independent Instance Maps for Crowd Localization"
MIT License
169 stars 41 forks source link

运行vis4val.py时出错 Key Error 求解~ #16

Open lizxko2009 opened 3 years ago

lizxko2009 commented 3 years ago

(IIM) E:\IIM-main\IIM>python vis4val.py 3110 Traceback (most recent call last): File "vis4val.py", line 101, in main() File "vis4val.py", line 34, in main if gt_data[i_sample]['num'] ==0 and pred_data[i_sample]['num'] !=0: KeyError: 3110

taohan10200 commented 3 years ago

请确认val_gt_loc.txt和生成的预测文件的格式，如下: pred.txt 3110+人数 + 预测的人头坐标(x, y) 3111 ....... ............... val_gt_loc.txt 3110 + 人数 +人头的位置和size信息(x, y, sigma_s, sigma_l, level) 3111 ...... ................. 其中，3110,3111对应着NWPU数据集里图片的ID