how to get the extract or get the trained model or weight file after running neural network model(specifically object detection yolo v2 model)

2 views (last 30 days)
Supreet bant
Supreet bant on 7 Jan 2022
Answered: Srivardhan Gadila on 11 Jan 2022

Answers (1)

Srivardhan Gadila
Srivardhan Gadila on 11 Jan 2022
Once you train the network using trainYOLOv2ObjectDetector, you can access the Network property of the yolov2ObjectDetector as follows:
dagnet = detector.Network
It would be a dagnetwork object and you can access the weights of the individual layers (layers having weights such as conv2d etc.) of the network as follows:
layer2 = dagnet.Layers(2)
weights2 = layer2.Weights;
You can also access them by checking the dagnet variable in the Workspace Browser as well.
Refer to the following pages for more information: yolov2ObjectDetector - properties, DAGNetwork.

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by