как добавить тензорный идентификатор в пример информации, python tenorflow - PullRequest
Новая
       33

как добавить тензорный идентификатор в пример информации, python tenorflow

0 голосов
/ 09 апреля 2020

Я строю набор данных, который имитирует структуру Mnist, но я не знаю, как добавить значение "id" к ключу изображения и метки. Я имитирую список наборов данных следующим образом: 

def parse_function(filename, label):

        image_string = tf.io.read_file(filename)
        image_decoded = tf.image.decode_jpeg(image_string)
        image_resized = tf.image.resize(image_decoded, [100, 100], method=tf.image.ResizeMethod.NEAREST_NEIGHBOR)
        data_out = {'image': image_resized, 'label': label}

        return data_out

def dataset_maker(list_sample_paths, list_labels):

        filenames = tf.constant(list_sample_paths)
        labels = tf.constant(list_labels)

        dataset = tf.data.Dataset.from_tensor_slices((filenames, labels))
        dataset = dataset.map(parse_function)

        return dataset

training_dataset = dataset_maker(list_training_sample_paths, list_training_sample_labels)
test_dataset = dataset_maker(list_test_sample_paths, list_test_sample_labels)

Вот значение случайного ключа "image" из набора данных Mnist: 

<tf.Tensor: id=140275, shape=(28, 28, 1), dtype=uint8, numpy=array([[[  0],[  0],[  0]], dtype=uint8)>

А вот значение случайный ключ «image» из моего набора данных: 

<tf.Tensor: shape=(28, 28, 1), dtype=uint8, numpy=array([[[  0],[  0],[  0]], >

Ну, во-первых, тип всплывает дважды в mnist (не знаю, почему)

Но самое главное, как я могу автоматически назначить идентификатор для образец в этом процессе?

Спасибо!

...