Как загрузить пользовательский набор данных в капсульную сеть?

Question

Я новичок в области глубокого обучения и его основ.Я собрал несколько изображений jpg велосипеда и объекта велосипеда.Теперь я хотел тренировать эти изображения, используя капсульную сеть.

Капсульная сеть реализована в наборе данных mnist.Код здесь .

Какие изменения необходимы для применения моего пользовательского набора данных.Изображение должно быть изменено на формат numpy или формат tfrecord?И как их загрузить по сети?

Answer 1

Чтобы загрузить пользовательский набор данных, я сослался на this .Это помогает.Набор данных преобразуется в массив NumPy и используется фон Keras.

Я загружаю пользовательский набор данных, а для модели капсульной сети используйте this .