- 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?
조회 수: 10 (최근 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개)
Nihal
2024년 9월 25일 7:04
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!