Преобразование тензор потока.python.data.ops.dataset_ops.DatasetV1Adapter в наборы данных тензор потока - PullRequest
Новая
       29

Преобразование тензор потока.python.data.ops.dataset_ops.DatasetV1Adapter в наборы данных тензор потока

0 голосов
/ 11 июля 2019

Я пытаюсь разработать модель классификации изображений с использованием CNN на своем собственном наборе данных изображений. Мне известно о разработке таких моделей с существующими наборами данных в тензорном потоке. Но я пытаюсь добавить свой набор данных в тензорный поток. Я уже пытался преобразовать подготовить изображениенабор данных, использующий приведенный ниже код и тип (набор данных): tenorflow.python.data.ops.dataset_ops.DatasetV1Adapter Теперь я хочу добавить это в наборы данных tenorflow, чтобы я мог легко загрузить набор данных с помощью dataset.load_data ()

Кто-нибудь может мне помочь?

В файле file.csv столбец Image содержит имена изображений, а столбец Class содержит класс изображений, которые я передаю эти столбцы в качестве параметров в tf.constant (), как показанов коде ниже. 

import pandas as pd
import tensorflow as tf

df = pd.read_csv('file.csv')

filenames = tf.constant(df.Image)
labels = tf.constant(df.Class)
dataset = tf.data.Dataset.from_tensor_slices((filenames, labels))

def _parse_function(filename, label):
    image_string = tf.read_file(filename)
    image_decoded = tf.image.decode_jpeg(image_string, channels=3)
    image = tf.cast(image_decoded, tf.float32)
   return image, label

dataset = dataset.map(_parse_function)
dataset = dataset.batch(2)

iterator = dataset.make_one_shot_iterator()
images, labels = iterator.get_next()
...