比较与tf.keras.datasets.cifar10的功能差异
tf.keras.datasets.cifar10
class tf.keras.datasets.cifar10()
更多内容详见tf.keras.datasets.cifar10。
mindspore.dataset.Cifar10Dataset
class mindspore.dataset.Cifar10Dataset(
dataset_dir,
usage=None,
num_samples=None,
num_parallel_workers=None,
shuffle=None,
sampler=None,
num_shards=None,
shard_id=None,
cache=None
)
使用方式
TensorFlow:可使用类内的 load_data
方法下载并加载Cifar10数据集。
MindSpore:从指定路径加载Cifar10数据集文件,并返回数据集对象。
代码示例
# The following implements Cifar10Dataset with MindSpore.
import mindspore.dataset as ds
cifar10_dataset_dir = "/path/to/cifar10_dataset_directory"
dataset = ds.Cifar10Dataset(dataset_dir=cifar10_dataset_dir)
# The following implements cifar10 with TensorFlow.
import tensorflow as tf
(x_train, y_train), (x_test, y_test) = tf.keras.datasets.cifar10.load_data()