在 windows 上安装 waymo 开放数据集

Question

我正在尝试在 windows 环境中安装 waymo 开放数据集。但是，当我尝试时，出现以下错误：

(python3.6) C:\Users\19729>pip install waymo-open-dataset-tf-2-3-0 ERROR: Could not find a version that satisfies the requirement waymo-open-dataset-tf-2-3-0 (from versions: none) ERROR: No matching distribution found for waymo-open-dataset-tf-2-3-0

如何解决？

Answer 1

当我在 colab 中使用 TensorFlow 2.7 进行尝试时，这段代码没有显示任何错误。

但是您可以使用以下代码重试：

pip3 install --upgrade pip
pip3 install waymo-open-dataset-tf-2-6-0 --user  # for tf 2.6.0.  OR
pip3 install waymo-open-dataset-tf-2-5-0 --user  # for tf 2.5.0.  OR
pip3 install waymo-open-dataset-tf-2-4-0 --user  # for tf 2.4.0.

您也可以使用以下方法从tensorflow_datasets 加载此built-in tensorflow dataset：

!pip install tensorflow_datasets

import tensorflow_datasets as tfds
from tensorflow_datasets.object_detection import WaymoOpenDataset
ds = tfds.load('waymo_open_dataset', split='train', shuffle_files=True)

注意：这可能需要一些时间，因为它是336.62 GiB 数据集。

在Kaggle 中也有一些waymo_open_datasets。