Вес файла использовать для изображения разного размера

Question

Я хочу обучить CNN, где размер изображения составляет 128 * 512, затем я хочу использовать этот файл весов для обучения других данных, которые имеют размерность 128 * 1024. Это означает, что я хочу использовать предварительно обученный файл веса во время тренировки различных данных (128 * 1024).

Это возможно или как я могу это сделать?

Я хочу сделать это, потому что у меня есть только 300 изображений с размером 128 * 1024, в то время как у меня есть 5000 изображений с размером 128 * 512, и оба набора данных отличаются.

Спасибо

Answer 1

Если ваша модель полностью CNN, совершенно не обязательно иметь разные модели.

Модель CNN может делать снимки любого размера.Просто убедитесь, что input_shape=(None,None,channels)

Вам понадобятся отдельные массивы, хотя один для больших изображений, другой для маленьких изображений, и вам придется вызывать разные методы подбора для каждого.

(но, вероятно, вы все равно будете работать с генератором)