Concatenate SSCB results from image layer with numeric CB feature data in deep learning model.
1 次查看(过去 30 天)
显示 更早的评论
I'm using trainNetwork() for transfer learning. I've set up a concatenation layer for combining metadata as an auxilary input with our image data to help with classifications.
I'm getting the following error.
Invalid network.
Caused by:
Layer 'concat': Input size mismatch. Size of input to this layer is different from the expected input size.
Inputs to this layer:
from layer 'prob' (size 1(S) × 1(S) × 8(C) × 1(B))
from layer 'features' (size 10(C) × 1(B))
How can I address this?
0 个评论
回答(1 个)
yanqi liu
2021-12-2
yes,sir,is the classes number 10?may be use
layersTransfer = net.Layers(1:end-3);
numClasses = 10;
layers = [
layersTransfer
fullyConnectedLayer(numClasses,'WeightLearnRateFactor',20,'BiasLearnRateFactor',20)
softmaxLayer
classificationLayer];
0 个评论
另请参阅
类别
在 Help Center 和 File Exchange 中查找有关 Image Data Workflows 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!