I want to use vgg16 for grayscale images with one channel

9 次查看（过去 30 天）
Aamir Shabbir hussain
评论： abhi abhi 2020-12-28
Hi Guyz.......i need to use CNN VGG16 for training my image samples but my images are gray scale and the input layer of VGG16 alexnet and VGG19 has 3 channels what can i do to resolve this issue. Please let me know some one.
Thanks,

请先登录，再回答此问题。

采纳的回答

Darshan Ramakant Bhat
Since VGG16 is a pretrained model its input configuration cannot be changed.You can copy the first Chanel values to other two channel and create a 3 channel image out of your gray scale image.
rgbImage = cat(3, grayImage, grayImage, grayImage);
Give this image as the input to VGG16. I hope this will workout for you.
  3 个评论
显示 1更早的评论
Image Analyst
Image Analyst 2017-11-19
It simply allocates more memory for rgbImage. It should be extremely fast and the computation time should not really be a concern.
Modar Alfadly
Modar Alfadly 2018-5-18
You can do the same as suggested more efficiently by summing up the channels of the kernels of the first convolutional layer.

请先登录，再进行评论。

更多回答（0 个）

类别

在 Help Center 和 File Exchange 中查找有关 Deep Learning Toolbox 的更多信息

标签

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by