如何使用自己较小的数据集运行代码？

Question

我想运行https://github.com/TropComplique/shufflenet-v2-tensorflow 用我自己的较小数据集。我该怎么办？!!!!

Answer 1

在train.py中：

PARAMS = {
    'train_dataset_path': '/mnt/datasets/imagenet/train_shards/',
    'val_dataset_path': '/mnt/datasets/imagenet/val_shards/',

}

更改数据集的路径