- 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?
7 次查看(过去 30 天)
显示 更早的评论
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 个评论
回答(1 个)
Yash Sharma
2024-7-4
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 个评论
另请参阅
类别
在 Help Center 和 File Exchange 中查找有关 Image Data Workflows 的更多信息
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!