Используйте предварительно обученный ResNet 50 в Keras для изображений с 4 каналами - PullRequest
0 голосов
/ 03 ноября 2018

Можно ли как-нибудь использовать веса ImageNet для ResNet50 для моего проекта с изображениями формы (224,224,4)? Изображение имеет R, G, B, Y каналы.

На данный момент я просто использую

 model = ResNet50(include_top=True, weights=None, input_tensor=None, input_shape=input_shape, pooling=None, classes=num_classes)

Теперь, если мне нужно использовать веса ImageNet, мне нужно всегда устанавливать количество классов равным 1000. Я попытался сделать это, а затем вытолкнуть последний слой и добавить свой собственный плотный (num_classes) слой. Однако теперь количество каналов является проблемой.

Кто-нибудь может предложить способ размещения 4 каналов в модели при использовании весов ImageNet?

...