I want to use vgg16 for grayscale images with one channel

25 次查看(过去 30 天)
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 个评论
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 CenterFile Exchange 中查找有关 Image Data Workflows 的更多信息

Community Treasure Hunt

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

Start Hunting!

Translated by