centerfusion网络中的数据集和参数问题

您好!我想询问一下centerfusion网络跑完整的数据集不下载mini数据集可以吗?就是直接下载10个部分+test部分的关键帧+相机+雷达数据。 我是否还需要修改src/tools/convert_nuScenes.py 文件中的NUM_SWEEPS=6参数? 我跑mini数据集时显示双卡3090和单卡3090 相差时间不多,我是否需要单卡训练? 还有一开始的预训练模型../models/centernet_baseline_e170.pth 我是否可以不需要?
非常感谢您的回答。祝一切顺利。

您好!感谢您的提问。

  1. 如果您想要跑完整的数据集,可以直接下载10个部分+test部分的关键帧+相机+雷达数据,不需要下载mini数据集。但是需要注意,完整数据集的大小会比mini数据集大很多,需要更多的存储空间和计算资源。

  2. 如果您使用的是完整数据集,可以不修改convert_nuScenes.py文件中的NUM_SWEEPS参数。该参数是用来控制每个样本中雷达数据的历史帧数,如果您使用的是完整数据集,每个样本中都包含了10帧雷达数据,因此不需要修改该参数。

  3. 如果您的双卡3090和单卡3090相差时间不多,可以考虑使用单卡训练。但是需要注意,单卡训练可能会导致训练时间变长,需要更多的时间和计算资源。

  4. 如果您不需要使用预训练模型,可以不使用../models/centernet_baseline_e170.pth。但是需要注意,如果您不使用预训练模型,需要更多的训练时间和计算资源来训练模型。

希望我的回答能够帮助到您,祝您一切顺利!