Как изменить этот Google Colab, чтобы он принимал мой собственный архивированный файл, и запускал модель на моих собственных изображениях?

Question

вот ссылка на блокнот Google Colab: https://colab.research.google.com/drive/1CwknSSXVtb-UBAw7Lm9slu_DgCO5M5wW

У меня проблемы с привязкой набора данных к коду и запуском модели на нем. Я пытаюсь адаптировать этот ноутбук https://colab.research.google.com/github/tensorflow/hub/blob/master/examples/colab/tf2_image_retraining.ipynb, чтобы запустить его на своем собственном наборе данных, но я получаю эту ошибку:

<ipython-input-14-542fb7cdb61e> in <module>()
      6     **datagen_kwargs)
      7 valid_generator = valid_datagen.flow_from_directory(
----> 8     data_dir, subset="validation", shuffle=False, **dataflow_kwargs)
      9 
     10 do_data_augmentation = False #@param {type:"boolean"}

1 frames
/tensorflow-2.1.0/python3.6/keras_preprocessing/image/directory_iterator.py in __init__(self, directory, image_data_generator, target_size, color_mode, classes, class_mode, batch_size, shuffle, seed, data_format, save_to_dir, save_prefix, save_format, follow_links, subset, interpolation, dtype)
    104         if not classes:
    105             classes = []
--> 106             for subdir in sorted(os.listdir(directory)):
    107                 if os.path.isdir(os.path.join(directory, subdir)):
    108                     classes.append(subdir)

NotADirectoryError: [Errno 20] Not a directory: '/root/.keras/datasets/training_photos'

, если кто-нибудь может дать мне несколько советов о том, как адаптировать исходную записную книжку google colab для запуска на моем собственном наборе данных (хранящемся в файлах моего компьютера) и получения прогнозов, я был бы очень благодарен, поскольку я новичок в этом.