在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()
好了,接下来训练就完美进行了。
版权声明:本文为m0_57213528原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。