- 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 Ansichten (letzte 30 Tage)
Ältere Kommentare anzeigen
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 Kommentare
Antworten (1)
Yash Sharma
am 4 Jul. 2024 um 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 Kommentare
Siehe auch
Kategorien
Mehr zu Image Data Workflows finden Sie in Help Center und File Exchange
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!