- 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 visualizaciones (últimos 30 días)
Mostrar comentarios más antiguos
Fahmi Akmal Dzulkifli
el 7 de Feb. de 2023
Respondida: Yash Sharma
el 4 de Jul. de 2024 a las 5:44
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 comentarios
Respuestas (1)
Yash Sharma
el 4 de Jul. de 2024 a las 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 comentarios
Ver también
Categorías
Más información sobre Image Data Workflows en Help Center y File Exchange.
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!