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 次查看(过去 30 天)
Supreet bant
Supreet bant2022-1-7

回答(1 个)

Srivardhan Gadila
Srivardhan Gadila 2022-1-11
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