Computer Vision Toolbox Model for Mask R-CNN Instance Segmentation is not working.

2 views (last 30 days)
I'm attempting to train the Mask R-CNN on a custom dataset. Mathworks has provided the toolbox for implementing Mask R-CNN. However, when I train the network with my custom dataset, I receive the following error:
[detector,info] = trainMaskRCNN(trainingData,lgraph,options,FreezeSubNetwork="backbone");
Invalid argument at position 1. Expected input to be of size MxNx5, but it is of size 500x724.
I'm not sure what this mistake means. I need clarification and assistance in determining what went wrong here.

Answers (1)

Prince Kumar
Prince Kumar on 6 Apr 2022
Hi,
The issue here is with your training data dimensions.
As it is clearly stated in the error "Expected input to be of size MxNx5, but it is of size 500x724". There is a dimension mis-match. Please have a look at you data more carefully.
Please go through the following documentation to have a better understanding:
Hope this helps!

Community Treasure Hunt

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

Start Hunting!