File Exchange

image thumbnail

Transfer Learning

version 5.1 (907 KB) by Kevin Chng
Transfer Learning of Pre-trained Neural Network or Imported ONNX Classification Model in GUI

250 Downloads

Updated 30 Jul 2019

View License

It allows user to do transfer learning of pre-trained neural network, imported ONNX classification model or imported MAT file classification model in GUI without coding. Other than using the existing model, user can design their neural network using Deep Network Designer (MATLAB built-in application) and later use this app to train the neural network. User can analyze the pre-trained network or imported network in this app. It also allows user to modify the image augmenter information and training option (hyperparameters). Once completed the training of network, user may export the trained network to workspace, MAT file, ONNX file and generate MATLAB code for the steps being done in the application. User can generate NVIDIA GPU CUDA Code, however it is not supporting imported ONNX model.

Pre-trained deep neural network available in this app:
01) alexnet
02) vgg16
03) vgg19
04) squeezenet
05) googlenet (ImageNet)
06) googlenet (Places365)
07) inceptionv3
08) densenet201
09) mobilenetv2
10) resnet18
11) resnet50
12) resnet101
13) xception
14) inceptionresnetv2
15) shufflenet
16) nasnetmobile
17) nasnetlarge
18) *import onnx model

How to import ONNX classification model for this application :
https://www.youtube.com/watch?v=8pZiduqp35g&feature=youtu.be

How to design a neural network and train the neural network in GUI :
Deep Network Designer + Transfer Learning (This App)
https://www.youtube.com/watch?v=-GeZa6IL2QA&feature=youtu.be

* New Feature :
1) Allow user to import ONNX classification model
* ONNX - Open Neural Network Exchange Format
2) Allow user to import model designed from MATLAB Deep Network Designer

Cite As

Kevin Chng (2019). Transfer Learning (https://www.mathworks.com/matlabcentral/fileexchange/70293-transfer-learning), MATLAB Central File Exchange. Retrieved .

Comments and Ratings (3)

Updates

5.1

Change Description

5.0

Allow user to import customized model designed from MATLAB Deep Network Designer

4.4

Change description

4.3

* Add video to demonstrate how to import ONNX model for this application.

4.2

Missing Picture

4.1

*change description

4.0

New Feature : Allow to import ONNX Classification Model for transfer learning, however, the imported ONNX model is not supported to generate GPU CUDA Code in this app.

3.4

Add more pre-trained neural network : xception,shufflent,nasnetmobile,nasnetlarge

3.3

Small Update (delete old result when start new session)

3.2

New Feature : Allow to change the learning rate of new last learnable layer

3.1

fix bug

3.0

Add new features:
1) Additional pre-trained neural networks : googlenet (Places 365), densenet201,mobilenetv2
2) More information about pre-trained neural network in display : Size, Number of Parameters (Millions)
3) Generate NVIDIA GPU CUDA Code

2.0

Change GUI

1.06

Change Title

1.05

Add the link of documentation

1.04

Update

1.03

*Add documentation to explain the apps

1.02

Change Description

1.1

*Change Description

MATLAB Release Compatibility
Created with R2019a
Compatible with any release
Platform Compatibility
Windows macOS Linux