У меня есть данные изображения в виде массива 4D, который содержит 841 изображение, каждое из которых имеет размер 56x56 пикселей и имеет 33 "цветных" канала. Теперь у меня есть один массив 4D для изображений с низким разрешением (который подвергается понижающей дискретизации по сравнению с изображениями с высоким разрешением) и один массив 4D для изображений с высоким разрешением, которые должны служить основной правдой для функции потерь. Мой вопрос сейчас состоит в том, как мне создать набор данных с этими массивами, чтобы у меня были пары изображений LR-HR, которые работают со следующим кодом (который я использовал, когда первоначально пробовал его с Div2K):
train = ... #dataset with HR-LR pairs
train_ds = train.dataset(batch_size=16, random_transform=True)
model.compile(optimizer='adam', loss='mean_absolute_error')
model.fit(train_ds, epochs=300, steps_per_epoch=1000)