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

12 次查看(过去 30 天)

回答(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.

类别

Help CenterFile Exchange 中查找有关 Deep Learning Toolbox 的更多信息

Community Treasure Hunt

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

Start Hunting!

Translated by