- removeLayers: https://www.mathworks.com/help/deeplearning/ref/nnet.cnn.layergraph.removelayers.html
- addLayers: https://www.mathworks.com/help/deeplearning/ref/nnet.cnn.layergraph.addlayers.html
How to modify the fully connected layer for YOLO purposes?
9 visualizzazioni (ultimi 30 giorni)
Mostra commenti meno recenti
Greetings
I have two classes and I want to use the Squeezenet network for YOLOv3 purposes. However, the last three layers in the network ('pool10', 'prob', and 'ClassificationLayer_predictions') have an ouput of 1x1x1000. How can I change the network for my purpose?
0 Commenti
Risposte (1)
Yash Sharma
il 4 Lug 2024 alle 5:44
Hi Fahmi,
You can use MATLAB's removeLayers function in order to remove a layer from a neural network, then you can add your own custom layer with the given output size of 1x1x1000 using addLayers function.
Hope it helps!
0 Commenti
Vedere anche
Categorie
Scopri di più su Image Data Workflows in Help Center e File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!