在PointPillar训练的阶段,出现了一个小bug:也就是一个需要bool型但是代码却是一个byte的错,需要做如下的修改
在second.pytorch\second\pytorch\models
文件夹下的voxelnet.py
的911行:
opp_labels = (box_preds[..., -1] > 0) ^ dir_labels.byte()
改为
opp_labels = (box_preds[..., -1] > 0) ^ dir_labels.bool()
好了,接下来训练就完美进行了。