According to the instructions in the README, I encountered a ValueError: Buffer dtype mismatch, expected 'int' but got 'long' in train.py at the 20th epoch. #4
Description
Traceback (most recent call last):
File "/home/lab/libr/WakeNet/eval.py", line 100, in
evaluate(arg.target_size,
File "/home/lab/libr/WakeNet/eval.py", line 85, in evaluate
results = data_evaluate(model, target_size, test_path, conf, dataset)
File "/home/lab/libr/WakeNet/eval.py", line 43, in data_evaluate
dets = im_detect(model, im, target_sizes=target_size, conf=conf)
File "/home/lab/libr/WakeNet/utils/detect.py", line 12, in im_detect
return single_scale_detect(model, src, target_size=target_sizes[0], use_gpu=use_gpu, conf=conf)
File "/home/lab/libr/WakeNet/utils/detect.py", line 36, in single_scale_detect
scores, classes, boxes, landmarks = model(im, test_conf=conf)
File "/opt/anaconda3/envs/libr/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1130, in _call_impl
return forward_call(*input, **kwargs)
File "/home/lab/libr/WakeNet/models/WakeNet.py", line 106, in forward
return self.decoder(imgs, anchors_list[-1], cls_score, bbox_pred, land_pred, test_conf=test_conf)
File "/home/lab/libr/WakeNet/models/WakeNet.py", line 124, in decoder
anchors_nms_idx = nms(torch.cat([bboxes.float(), scores.float()], dim=2)[0, :, :], nms_thresh)
File "/home/lab/libr/WakeNet/utils/nms_wrapper.py", line 13, in nms
return cpu_nms(dets, thresh) # 加了个int
File "cpu_nms.pyx", line 153, in cpu_nms.cpu_nms
ValueError: Buffer dtype mismatch, expected 'int' but got 'long'