Cannot import trained network

조회 수: 7 (최근 30일)
yongbo chen
yongbo chen 2021년 6월 5일
댓글: yongbo chen 2021년 6월 19일
I trained a network using tensorflow 2.0 in windows 10. It works fine and can output the predicted result. However, when I try to use "net = importKerasNetwork('model.h5')", MATLAB has the following error:
Error using assembleNetwork (line 47)
Invalid network.
Error in nnet.internal.cnn.keras.importKerasNetwork (line 35)
Network = assembleNetwork(LayersOrGraph);
Error in importKerasNetwork (line 91)
Network = nnet.internal.cnn.keras.importKerasNetwork(modelfile, varargin{:});
Caused by:
Network: Missing output layer. The network must have at least one output layer.
Layer 'conv2d_8_OutputLayer_PLACEHOLDER': Unconnected output. Each layer output must be connected to the input
of another layer.
Layer 'conv2d_8_OutputLayer_PLACEHOLDER': Layer validation failed. Error using 'forward' in Layer
nnet.keras.layer.PlaceholderOutputLayer. The function threw an error and could not be executed.
Error using nnet.internal.cnn.layer.util.CustomLayerLegacyStrategy/forward (line 42)
Networks containing PlaceholderLayers cannot be trained or used for prediction. Either remove or replace
all PlaceholderLayers.
Any possible way to revise it? Thanks a lot.

답변 (2개)

Divyam Gupta
Divyam Gupta 2021년 6월 8일
Hi Yongbo,
I understand that you're having an issue in importing a trained network. The error that I notice is that your network is missing an output layer. Please refer to the following answer on MATLAB Central which discusses a similar issue that you are facing: https://www.mathworks.com/matlabcentral/answers/673738-importkerasnetwork-for-custom-loss-function
Hope this helps.

Abhishek Gupta
Abhishek Gupta 2021년 6월 9일
Hi,
As per my understanding, you are getting an error while importing a network.
If your network doesn't have a loss layer, then you need to import your network using importKerasLayers and add a custom output layer at the end, or you can convert the layerGraph to dlnetwork.
Referring to the following documentation, which might help you in resolving the issue: -
  댓글 수: 3
Abhishek Gupta
Abhishek Gupta 2021년 6월 15일
You are getting this error because the second argument of your predict function is not of type dlarray. For more information related to the input data, check out this documentation link: -https://www.mathworks.com/help/deeplearning/ref/dlnetwork.predict.html#function_dlnetwork_sep_predict_sep_mw_bfae26e4-e62e-4b46-80e2-3d3f8305b520
yongbo chen
yongbo chen 2021년 6월 19일
Thank you very much for your answer. I have revised the input and get the predict result. However, I find that the output of the network is totally different from the result from tensor flow using python. The trained network "lgraph" from tensorflow 2.4.1. Any possible way to solve this problem:
lgraph = importTensorFlowLayers('D:\models');
I = imread('D:\1.jpg');
dlX = dlarray(single(I),'SSCB');
dlnet = dlnetwork(lgraph);
dlY = predict(dlnet,dlX);
x = extractdata(dlY);

댓글을 달려면 로그인하십시오.

카테고리

Help CenterFile Exchange에서 Image Data Workflows에 대해 자세히 알아보기

제품


릴리스

R2020b

Community Treasure Hunt

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

Start Hunting!

Translated by