当运行second网络中train.py时可能会出现以下错误:
可以看到第871行时报错的根本原因
File "./pytorch/train.py", line 871, in predict_v2
opp_labels = (box_preds[..., -1] > 0) ^ dir_labels.byte()
报错原因:
这是SECOND-V1.5是用较老的torch版本写的,你的torch版本可能太新了,你可以选择换个torch版本,但是大多数人一旦换版本可能会出现其他奇奇怪怪的问题。
解决方法:
所以我们需要将871行中的代码换成旧版本torch也可以使用的格式。我们就给他加上bool值。
将
change opp_labels = (box_preds[..., -1] > 0) ^ dir_labels.byte()
改为:
opp_labels = (box_preds[..., -1] > 0) ^ dir_labels.to(torch.bool)
错误解决,Nice!