@zeni1agent

Как подготовить данные для генератора?

Я получаю значения из каталога

dateset = image_dataset_from_directory(
                            'Fruit-Images-Dataset-master/Training',
                            subset='training',
                            seed=42,
                            validation_split=0.1,
                            image_size=image_size
)

И я хочу отправить на генератор и дискриминатор набор из вот таких данных
[[image][tags][title]] 
   # image = [0.99607843, 1.         ,0.99215686, ...] 
   # tags = [Tomato, Tomato Cherry Red]
   # title = 1.png

Но проблема в том что я не понимаю как взаимодействовать с дата сетом

Я пытался каталогизировать изображения так

Training
   Tomato
      Tomato 1
        1.png
          ...
       Tomato Cherry Red
         1.png
         2.png
            ...

И вместо PrefetchDataset выводит BatchDataset
А class_names выводит только Tomato а Tomato Cherry Red не видит

Как сделать так что бы метки было больше чем одна?

Я пытался пройтись по каждому элементу отдельно

SIZE = 224
def resize_image(img, label):
  img = img.numpy().astype("uint8")
  img = tf.cast(img, tf.float32)
  img = tf.image.resize(img, (SIZE, SIZE)) # изменить размер
  img = img / 255.0 # Перемножить значение что бы оно было равно 0. или 1.
  img = np.concatenate(myimg, axis=1) # Соединить 3 цвета в одну строку 
  return img, label
train_resized = training_dateset[0].map(resize_image) # Преобразовать

Но получил

'PrefetchDataset' object is not subscriptable?

Хотя у меня BatchDataset
с PrefetchDataset из tfds.load почему-то работает без проблем

Как мне правильно подготовить данные(Редактировать их)? Что бы их структура была примерно такой

[
    [image][tags][title]
    [image][tags][title]
    [image][tags][title]
  ]
  • Вопрос задан
  • 112 просмотров
Пригласить эксперта
Ваш ответ на вопрос

Войдите, чтобы написать ответ

Похожие вопросы