是否有人尝试过使用“NuScenes数据集”？

Question

我正在尝试执行一个示例，CameraRadarFusionNet。 （https://github.com/TUMFTM/CameraRadarFusionNet）

我在Ubuntu上做了“git克隆”， 并执行了以下教程。 我还下载了所需的数据集

我犯了这个错误

File "../crfnet/data_processing/generator/crf_main_generator.py", line 78, in create_generators
    nusc = NuScenes(version='v1.0-mini', dataroot='/data/sets/nuscenes/', verbose=True)
  File "/home/aicar/.local/lib/python3.6/site-packages/nuscenes/nuscenes.py", line 54, in __init__
    assert osp.exists(self.table_root), 'Database version not found: {}'.format(self.table_root)
AssertionError: Database version not found: /data/sets/nuscenes/v1.0-mini

怎么了

我有 Ubuntu 18.04 python 3.6 cuda 10.0，cudnn，tensorflow gpu

Answer 1

我在使用nuScenes数据集时遇到过类似问题，以下是我的解决方案：

一,。下载并解压缩数据集

您需要在v1.0-mini（或者v1.0-trainval，或者v1.0-test中解压缩您的内容，如果您下载了其他两个版本）。您应该看到与此类似的文件结构：

picture of my file structure

二,。更改dataroot和version的默认值：

然后，转到~/nuscenes-devkit/python-sdk/nuscenes/scripts/assert_download.py文件，它最初应该是这样的（主要部分）：

if __name__ == "__main__":

# Settings.
parser = argparse.ArgumentParser(description='Test that the installed dataset is complete.',
                                 formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument(' dataroot', type=str, default='/home/ss/data/sets/nuscenes',
                    help='Default nuScenes data directory.')
parser.add_argument(' version', type=str, default='v1.0-trainval',
                    help='Which version of the nuScenes dataset to evaluate on, e.g. v1.0-trainval.')

将dataroot更改为绝对路径，并将version更改为刚下载的数据集的版本。例如，我的看起来像：

parser.add_argument(' dataroot', type=str, default='/home/ss/data/sets/nuscenes',
                    help='Default nuScenes data directory.')
parser.add_argument(' version', type=str, default='v1.0-mini',
                    help='Which version of the nuScenes dataset to evaluate on, e.g. v1.0-trainval.')

那你就可以走了

三,。验证您的设置

在端子中，设置环境。变量：

export NUSCENES="/data/sets/nuscenes"

然后在目录~/nuscenes-devkit/python-sdk/nuscenes/scripts中运行：

python3 assert_download.py

如果v1.0-mini数据集的设置正确，您应该能够看到成功

希望我的回答对你有帮助