Могу ли я пакетный набор данных дважды? - PullRequest
Новая
       8

Могу ли я пакетный набор данных дважды?

0 голосов
/ 26 февраля 2019

Я пытаюсь группировать набор данных изображений в группы по 4. Каждая партия из 4 изображений используется для создания нового отдельного изображения.Наконец, я хотел бы объединить все полученные составные изображения в пакеты 512. Могу ли я сделать что-то вроде этого? 

def read_dataset(filename, mode, batch_size = 512):
    def _input_fn():

        ...

        # Parse text lines as comma-separated values (CSV)
        dataset = textlines_dataset.apply(
        tf.data.experimental.map_and_batch(decode_csv,num_parallel_calls=os.cpu_count(),batch_size = 4))

        dataset = dataset.apply(
        tf.data.experimental.map_and_batch(augment4,num_parallel_calls=os.cpu_count(),batch_size = batch_size))


        dataset = dataset.prefetch(buffer_size=None)

        return dataset
    return _input_fn
...