загрузка данных tenorflow для нескольких входов - PullRequest
0 голосов
/ 09 мая 2020

Я пытаюсь обучить сеть с 3 входами, но у меня возникают проблемы с загрузкой данных тензорного потока. Я разработал загрузчик данных с помощью tensorflow.keras.utils.Sequence в соответствии с их документацией, но он очень медленный (50 с / шаг на моем процессоре) - также с очень простой моделью c одного уровня. Мой загрузчик данных читает 3 имени изображений из дейтаграммы, загружает их и изменяет их форму (код ниже). загрузка данных вместо keras.Sequence, но я не нашел способа выполнить такую ​​же загрузку данных с tf.data, чем с keras.Sequence.

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

изменить: фрейм данных, из которого считываются имена / пути изображений, выглядит следующим образом:

+--+-------+-------+-------+-------+
|  | A     | B     | C     | label |
+--+-------+-------+-------+-------+
|  | 00013 | 04704 | 04774 | 0     |
+--+-------+-------+-------+-------+
|  | 00037 | 00366 | 01986 | 0     |
+--+-------+-------+-------+-------+
|  | 02945 | 04979 | 01875 | 1     |
+--+-------+-------+-------+-------+
|  | 02214 | 00181 | 00872 | 1     |
+--+-------+-------+-------+-------+
|  | 04086 | 03791 | 03104 | 1     |
+--+-------+-------+-------+-------+
...