时间戳

首页 > 解决方案 > 有没有人尝试过使用“NuScenes 数据集”？

python - 有没有人尝试过使用“NuScenes 数据集”？

问题描述

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

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

我得到了这个错误。

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。

标签： pythondataset

解决方案

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

1. 下载并解压数据集

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

我的文件结构的图片

dataroot2. 更改and的默认值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.')

那你就可以走了！

3. 验证您的设置

在终端中，设置环境。多变的：

export NUSCENES="/data/sets/nuscenes"

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

python3 assert_download.py

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

希望我的回答对你有所帮助。

推荐阅读