Input to Convolution Neural Network

3 次查看(过去 30 天)
AP
AP 2020-6-29
I am using convolution neural network on face images of the size 1024*768*3.
I want to pass these images as input to first input layer of CNN.
what should be size of the 'inputImageLayer' for CNN?
How to decide the size of First layer of CNN?

回答(1 个)

bharath pro
bharath pro 2020-6-29
The ImageInputLayer is used to take image inputs when designing a CNN. According to the documentation, it can be called using ImageInputLayer([h w c]), where c= number of chanels. I am assuming for you the height and width of images are 1024 and 768 respectively and c=3. So for this you would need to use ImageInputLayer([1024 768 3]) to instantiate this layer.
  8 个评论
AP
AP 2020-6-30
Clc;
clear all;
close all;
%read image
a = imread('tryb.jpg');
size(a)
a1 =rgb2gray(a);
%show image
figure, imshow(a1);
imageSize =[768 1024 3];
matlabroot ='D:\MY DESKTOP\IMAGE PROCESSING MATERIAL\database\for nose detect\';
Datasetpath =fullfile(matlabroot ,'cnn2','Dataset1');
Data = imageDatastore(Datasetpath,'IncludeSubfolders',true,'LabelSource','foldernames');
layers = [
imageInputLayer (imageSize)
covolution2dLayer([5 20])
reluLayer
maxPooling2dLayer(2,'stride',2)
covolution2dlayer([5 20])
reluLayer
maxPooling2dLayer(2,'stride',2)
fullyConnectedLayer(2)
softmaxLayer
ClassiifcationLayer()];
options = trainingOptions('sgdm','maxEpoch',15,'initialLearningrate , 0.001');
convnet = trainNetwork (Data,layers,options);
output = classify (convnet ,a);
bharath pro
bharath pro 2020-6-30
change your layers to this:
layers = [
imageInputLayer(imageSize,'Normalization','none')
convolution2dLayer(5,20)
reluLayer()
maxPooling2dLayer(2,'Stride',2)
convolution2dLayer(5,20)
reluLayer()
maxPooling2dLayer(2,'Stride',2)
fullyConnectedLayer(1)
softmaxLayer
classificationLayer()];
Also your options seem to be wrong. Please change that.

请先登录,再进行评论。

类别

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