他的回复:
问题找到了,修改原文件fairmot_postprocess.py中:hm_eval = torch.from_numpy(np.fromfile(dataloader[i + 3], dtype='float32').reshape(1, 1, 152, 272))wh_eval = torch.from_numpy(np.fromfile(dataloader[i + 2],dtype='float32').reshape(1, 4, 152, 272))id_eval = torch.from_numpy(np.fromfile(dataloader[i + 1],dtype='float32').reshape(1, 128, 152, 272))reg_eval = torch.from_numpy(np.fromfile(dataloader[i],dtype='float32').reshape(1, 2, 152, 272))为:hm_eval = torch.from_numpy(np.fromfile(dataloader[i + 0], dtype='float32').reshape(1, 1, 152, 272))wh_eval = torch.from_numpy(np.fromfile(dataloader[i + 1],dtype='float32').reshape(1, 4, 152, 272))id_eval = torch.from_numpy(np.fromfile(dataloader[i + 2],dtype='float32').reshape(1, 128, 152, 272))reg_eval = torch.from_numpy(np.fromfile(dataloader[i + 3],dtype='float32').reshape(1, 2, 152, 272))就行了,就是dataloader中次序反了。修改后,输出结果:Fix size testing. training chunk_sizes: [6, 6] The output will be saved to ./FairMOT/src/lib/../../exp/mot/default heads {'hm': 1, 'wh': 4, 'id': 128, 'reg': 2} 2024-02-16 17:19:08 [INFO]: start seq: MOT17-02-SDP 2024-02-16 17:19:08 [INFO]: start seq: MOT17-02-SDP /pytorch/aten/src/ATen/native/BinaryOps.cpp:81: UserWarning: Integer division of tensors using div or / is deprecated, and in a future release div will perform true division as in Python 3. Use true_divide or floor_divide (// in Python) instead. 2024-02-16 17:19:09 [INFO]: save results to ./dataset/MOT17/images/train/../results/MOT17_test_public_dla34/MOT17-02-SDP.txt 2024-02-16 17:19:09 [INFO]: save results to ./dataset/MOT17/images/train/../results/MOT17_test_public_dla34/MOT17-02-SDP.txt 2024-02-16 17:19:09 [INFO]: Evaluate seq: MOT17-02-SDP 2024-02-16 17:19:09 [INFO]: Evaluate seq: MOT17-02-SDP IDF1 IDP IDR Rcll Prcn GT MT PT ML FP FN IDs FM MOTA MOTP IDt IDa IDm MOT17-02-SDP 91.3% 96.9% 86.4% 86.4% 96.9% 22 19 0 3 3 15 0 0 83.6% 0.163 0 0 0 OVERALL 91.3% 96.9% 86.4% 86.4% 96.9% 22 19 0 3 3 15 0 0 83.6% 0.163 0 0 0