Custom Mask-rcnn

Question

Claudia De Clemente 2023-5-5

0
链接

此问题的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1958579-custom-mask-rcnn

回答： Sachin 2023-5-15

Hello, is it possible to make a smaller mask rcnn? that is, to build it on a smaller res-net 50? when training YOLO, i can choose the feature extraction layer. Can I do it also for Mask rcnn? I don't really understand the literature, and the training of the net is taking so long that I was not able to complete a single training session, let alone try to tune the right hyperparameters for my application. I have a set of brigthfield images, [2048x2048], which I downsize to 300x300x3 for the training.

Thank you to whoever will respond.

0 个评论
显示 -2更早的评论隐藏 -2更早的评论

请先登录，再进行评论。

请先登录，再回答此问题。

Answer 1

Sachin 2023-5-15

0
链接

此回答的直接链接

https://ww2.mathworks.cn/matlabcentral/answers/1958579-custom-mask-rcnn#answer_1234964

在 MATLAB Online 中打开

Hi

I understand that you want to create a smaller mask RCNN. The following workaround might be of good help to you.

You can use ‘removelayers’ to remove a particular layers and “addlayers” to add a layers.

Here is an example which you can refer:

net = resnet50; % resnet object model
lgraph = layerGraph(net); % architecture of a deep learning network
lgraph = removeLayers(lgraph, {'fc1000', 'prob', 'ClassificationLayer_predictions'}); % to remove a layer
feature_layer = 'activation_40_relu'; % feature extraction layer
lgraph = connectLayers(lgraph, 'res5c_relu', feature_layer); % connect all the layers