[深度学习][CenterFusion]关于centerfusion训练注意的要点

我参考博客https://blog.csdn.net/ssj925319/article/details/124294911写得非常详细，但是有些地方需要注意

第一个地方：

scikit-learn版本问题，官方要求是0.21.0版本，我安装0.21.3这个版本也是可以的

python版本问题：我使用的python3.8版本没有任何问题，所以不仅仅是python3.7支持如果你是python3.8也是没问题的

pytorch版本问题：我使用的torch==1.11.0+cu113 torchvision==0.12.0+cu113版本，训练时候会报错torchvision.model.utils找不到，只要改成

即将 from torchvision.models.utils import load_state_dict_from_url 改为 from torch.hub import load_state_dict_from_url。

最后就是参数设置问题，注意gpu 0,1你有一个显卡把1去掉，batch-size不要太大不然容易爆显存。其他安装教程走就行了。

第二个地方：训练时候报错

TypeError: only integer tensors of a single element can be converted to an index

解决方法：

找到代码位置CenterFusion/src/lib/utils/pointcloud.py:

    bbox_int = torch.tensor([torch.floor(bbox[0]), 
                         torch.floor(bbox[1]), 
                         torch.ceil(bbox[2]), 
                         torch.ceil(bbox[3])], dtype=torch.int32)# format: xyxy

to

    bbox_int = torch.tensor([int(torch.floor(bbox[0])), 
                         int(torch.floor(bbox[1])), 
                         int(torch.ceil(bbox[2])), 
                         int(torch.ceil(bbox[3]))], dtype=torch.int32)# format: xyxy