how can I do object detection with vgg16 by using rcnn??
조회 수: 1 (최근 30일)
이전 댓글 표시
Hello, I have my custom image data of car and I labeled all my images.
And I want to train CNN by using rcnn. source code is 'rcnn = trainRCNNObjectDetector(gTruth, layers, options)
gTruth is my dataset and I want to use vgg16 for layers.
accodring by this url, https://kr.mathworks.com/help/deeplearning/ref/vgg16.html
I can't modify inputlayers(imagesize) and number of nodes of last classification layer and number of classess.
how can I do object detection with vgg16 by using rcnn??
댓글 수: 0
답변 (1개)
Sourav Bairagya
2020년 1월 10일
You can extract the desired number of layers from the pretrained net and define your choice of network. Here is an small example for extracting the desired layers.
net = vgg16;
lastFeatureLayerIdx = 32;
layers = net.Layers;
middlelayers = layers(2:lastFeatureLayerIdx);
Now, you can define your choice of input layer and final layer (i.e. classification layer) and then combine them with the middle layers to get the complete network.
layers = [
inputLayer
middleLayers
finalLayers
]
Now, you can leverage this link to get an idea how to perform object detection using R-CNN:
댓글 수: 0
참고 항목
Community Treasure Hunt
Find the treasures in MATLAB Central and discover how the community can help you!
Start Hunting!