VGG16 обучен на изображении в оттенках серого

Я обнаружил, что сеть VGG16 предварительно обучена в (цветной) базе данных imagenet (как .npy). Доступна ли сеть VGG16, предварительно обученная на серой версии базы данных imagenet?

(Обычных приемов использования 3-канальных фильтров слоя conv1.1 на сером 1-канальном входе мне недостаточно. Я смотрю на постепенное улучшение производительности сети, поэтому мне нужно посмотреть, как Трансферное обучение ведет себя, когда предварительно обученная модель «смотрит» на ввод в серой шкале).

Спасибо!


conv-neural-network pre-trained-model imagenet grayscale
person MathMits    schedule 05.02.2018    source источник
comment
Я не думаю, что есть что-то доступное. Вы можете самостоятельно обучить его работе с облачными сервисами, такими как Floydhub.   -  person Tien Dinh    schedule 07.02.2018
comment
Спасибо за ответ Люк. Сам пока не нашел. Я не знаю, смогу ли я получить достаточно ресурсов, чтобы сделать это сам. так я надеюсь, что кто-то уже сделал это.   -  person MathMits    schedule 19.03.2018
comment
Вы можете попробовать точную настройку предварительно обученного VGG16   -  person Tien Dinh    schedule 19.03.2018
comment
Да, Люк, это то, что я делаю, когда выполняю трансферное обучение (ты это имеешь в виду, верно?). Но в моем случае, как это обычно бывает, я считаю более эффективным оставить коэффициенты первых сверточных слоев неизменными (не обучаемыми). Тем не менее, я хотел бы, чтобы эти (первые слои) коэффициенты были предварительно обучены на огромной серой шкале изображений.   -  person MathMits    schedule 20.03.2018
comment
Любопытно услышать, пробовали ли вы заморозить все слои, кроме первого, во время тренировки. Затем, когда это станет стабильным, тренируйте всех.   -  person Alexander Soare    schedule 22.12.2020

Ответы (2)


arrow_upward
1
arrow_downward

Да, вот этот: https://github.com/DaveRichmond-/grayscale-imagenet

Обученная модель изображения в оттенках серого, а также ее версия, настроенная на рентгеновских снимках. Они показали, что производительность Imagenet практически не падает.

person GrimSqueaker    schedule 22.12.2020

arrow_upward
0
arrow_downward

@GrimSqueaker предоставил вам код этой статьи:

Однако обученная в нем модель - это Inception v3, а не VGG16.

У вас есть два варианта:

  1. Используйте цветную предварительно обученную модель VGG16 и продублируйте один канал на три канала.
  2. Обучите свою модель VGG16 на наборе данных ImageNet в оттенках серого.

Вы можете найти эту ссылку полезной: https://github.com/zzangho/VGG16_grayscale

person ma710u5    schedule 15.07.2021