File Exchange

image thumbnail

Transfer Learning

version 2019.6.2 (1.17 MB) by Kevin Chng
Transfer Learning of Pre-trained Neural Network or Imported ONNX Classification Model in GUI

284 Downloads

Updated 05 Nov 2019

View Version History

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 Neural Network available :
1) alexnet
2) googlenet(ImageNet)
3) goolgenet(Places365)
4) resnet18
5) resnet50
6) resnet101
7) vgg16
8) vgg19
9) inceptionv3
10) inceptionresnetv2
11) squeezenet
12) densenet201
13) mobilenetv2
14) shufflenet
16) xception
17) nasnetmobile
18) nasnetlarge
19) import ONNX model
20) import mat file model

[Up to R2019a, imported ONNX layer is not supporting generation of CUDA code]

How to design & train neural network using this app?
https://www.youtube.com/watch?v=-GeZa6IL2QA

How to import ONNX Model to train using this app?
https://www.youtube.com/watch?v=8pZiduqp35g

Cite As

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

Comments and Ratings (10)

Liu Fred

Anatoly Vasiliev

junhan park

Ian Alferez

Safwana Razak

Mahmoud Afifi

chiris leon

HIgh Tech Man

Safwana Razak

Ian Alferez

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

Community Treasure Hunt

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

Start Hunting!