필터 지우기
필터 지우기

How to combine two pre trained deep learning models (Mobilenetv2 and nasnet_mobile)

조회 수: 54 (최근 30일)
Hi
how can we create a hybrid model based on two pretrained models (Mobilenetv2 and nasnet_mobile) ?
Previously I did for mobilenetv2 and Resnet using a deep network analyzer app by replacing input layer and classification layers of both models with single input layer and classification layer with the help of a depth concatination layer. it worrks fine.
but when i try to do with the these two models it is not working..
any link available for proper way of combining models available?
the model it generted after the procedure is attached herewith
Thank You

채택된 답변

Milan Bansal
Milan Bansal 2024년 4월 4일
Hi Ravi,
I understand that you are trying to create a hybrid model based on two pretrained models (Mobilenetv2 and nasnet_mobile) and are facing issues with the workflow of creating a single input and classification layer using "depthConcatenationLayer".
Encountered issues might be due to differences in the input sizes, layer names, or the way features are extracted and concatenated. For this particualar case where "MobileNet-v2" and "NasNet-Mobile" needs to be combinen, Deep Network Designer can be used. Please follow the below steps to combine the two models.
  1. Open Deep Network Designer App by entering the command "deepNetworkDesigner" in the command window.
  2. Click on the "New" button at the top left corner.
  3. Select "NasNet-Mobile" from Image Networks.
  4. Repeat the previous step to add "MobileNet-v2" to same canvas as "NasNet-Mobile".
  5. Both the models expects an input image of size 224x224x3. Remove the image input layer from one of MobileNet-v2 and connect the remaining part to the image input layer of the NasNet-Mobile.
  6. Both the network have a layer with the name "global_average_pooling_2d_1"(3rd layer from the last in both the network), change the name of layers with this name as all the layer names must be unique in a Neural Network. Change the name to "global_average_pooling_2d_1_NasNet" for NasNet-Mobile and "global_average_pooling_2d_1_mobileNetv2" for MobileNet-v2.
  7. Remove the softmax prediction layer and the corresponding fully connected layer from both the networks.
  8. Add a "depthConcatenationLayer" and connect the output of "global_average_pooling_2d_1_NasNet" and "global_average_pooling_2d_1_mobileNetv2" to it.
  9. Finally add a "fullyConnectedLayer" and prediction layer to the "depthConcatenationLayer" added in the previous step.
  10. Analyze the network and check for errors.
Please refer to the following documentation link to learn more about Deep Network Designer App.
Hope this helps.

추가 답변 (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