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

我参考博客CenterFusion 项目超详细环境搭建步骤及可视化操作_centernet可视化_上班摸不了鱼的博客-CSDN博客本文引用 CenterFusion: Center-based Radar and Camera Fusion for 3D Object Detection，复现此论文中的代码，互相学习。注意：由于博主这几天都在复现这个项目，博客的内容的进度也是博主项目复现的进度，如果遇到找不到页面的情况，是由于内容在待审核，过一会儿就出来了。一、AutoDL云服务器说明：如果使用其它服务器，可以忽略掉这部分。目...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