Mask R-CNN overview in Computer Vision - Model Pipeline Trace

Model Pipeline - Mask R-CNN overview

Mask R-CNN is a model that finds objects in pictures, draws boxes around them, and colors each object precisely. It does this by first spotting objects, then figuring out their exact shape.

Data Flow - 6 Stages

1Input Image

1 image x 800 x 800 x 3 channels→Original color image loaded→1 image x 800 x 800 x 3 channels

A photo of a dog and a cat

↓

2Backbone CNN

1 image x 800 x 800 x 3→Extracts features like edges and textures→1 image x 50 x 50 x 256 feature maps

Feature maps highlighting shapes of dog and cat

↓

3Region Proposal Network (RPN)

1 image x 50 x 50 x 256→Suggests boxes where objects might be→1000 boxes x 4 coordinates + scores

Boxes around dog ears, cat face, etc.

↓

4RoI Align

1000 boxes x feature maps→Extracts fixed-size features for each box→1000 regions x 14 x 14 x 256

Features for each proposed box

↓

5Bounding Box & Classifier Head

1000 regions x 14 x 14 x 256→Predicts object class and refines box→1000 boxes with class labels and refined coordinates

Box labeled 'dog' with improved position

↓

6Mask Head

1000 regions x 14 x 14 x 256→Predicts pixel-level mask for each object→1000 masks x 28 x 28 pixels

Mask showing exact dog shape inside box

Training Trace - Epoch by Epoch

Loss: 1.2 |****     
Loss: 0.9 |******   
Loss: 0.7 |******** 
Loss: 0.55|*********
Loss: 0.45|**********

Epoch	Loss ↓	Accuracy ↑	Observation
1	1.2	0.45	Model starts learning basic object shapes
2	0.9	0.60	Better detection and mask prediction
3	0.7	0.72	Improved box accuracy and mask quality
4	0.55	0.80	Model refines object boundaries well
5	0.45	0.85	Strong detection and precise masks

Prediction Trace - 6 Layers

Layer 1: Input Image

Layer 2: Backbone CNN

Layer 3: Region Proposal Network

Layer 4: RoI Align

Layer 5: Bounding Box & Classifier Head

Layer 6: Mask Head

Model Quiz - 3 Questions

What does the Region Proposal Network (RPN) do in Mask R-CNN?

A Colors the objects in the image

B Extracts edges and textures

C Suggests boxes where objects might be

D Predicts the object class

Key Insight

Mask R-CNN combines object detection and segmentation by first finding object boxes and then predicting detailed masks, allowing precise understanding of objects in images.

Practice

1. What is the main purpose of Mask R-CNN in computer vision?

easy

A. To generate captions for images

B. To classify images into categories

C. To enhance image resolution

D. To detect objects and find their exact shapes in images

Solution

1. Step 1: Understand Mask R-CNN's function

   Mask R-CNN is designed to detect objects and also predict masks that show the exact shape of each object.

2. Step 2: Compare with other tasks

   Unlike classification or captioning, Mask R-CNN focuses on locating and segmenting objects precisely.

3. Final Answer:

   To detect objects and find their exact shapes in images -> Option D

4. Quick Check:

   Mask R-CNN = Object detection + precise shape masks [OK]

Hint: Mask R-CNN = detection + mask prediction [OK]

Common Mistakes:

• Confusing Mask R-CNN with image classification models
• Thinking it only detects bounding boxes without masks
• Assuming it generates image captions

2. Which component is NOT part of the Mask R-CNN architecture?

easy

A. Region Proposal Network (RPN)

B. Mask branch for pixel-level segmentation

C. Recurrent Neural Network (RNN) for sequence modeling

D. Fully Connected Layer for classification

Solution

1. Step 1: Identify Mask R-CNN components

   Mask R-CNN includes RPN for proposing regions, fully connected layers for classifying objects, and a mask branch for segmentation.

2. Step 2: Check for unrelated components

   RNNs are used for sequence data like text or speech, not for Mask R-CNN's image segmentation tasks.

3. Final Answer:

   Recurrent Neural Network (RNN) for sequence modeling -> Option C

4. Quick Check:

   Mask R-CNN excludes RNNs [OK]

Hint: Mask R-CNN uses CNNs, not RNNs [OK]

Common Mistakes:

• Thinking RNNs are part of Mask R-CNN
• Confusing mask branch with classification layers
• Ignoring the role of RPN

3. Given the following simplified Mask R-CNN output for an image, what does the mask represent?
masks = [[[0,1,1],[0,1,1],[0,0,0]], [[1,0,0],[1,1,0],[0,0,0]]]
Each inner list is a 3x3 mask for one object. What does a '1' indicate?

medium

A. Pixels belonging to the object

B. Background pixels

C. Pixels with low confidence

D. Pixels to be ignored

Solution

1. Step 1: Understand mask values in Mask R-CNN

   Mask R-CNN outputs masks where '1' marks pixels that belong to the detected object, and '0' marks background.

2. Step 2: Interpret the given mask arrays

   In the given masks, '1's show the shape of each object in the 3x3 grid.

3. Final Answer:

   Pixels belonging to the object -> Option A

4. Quick Check:

   Mask '1' = object pixel [OK]

Hint: '1' in mask means object pixel [OK]

Common Mistakes:

• Confusing '1' with background
• Assuming '1' means low confidence
• Thinking '1' means ignored pixels

4. You run Mask R-CNN but get an error: 'ValueError: mask size mismatch'. What is the most likely cause?

medium

A. The mask output size does not match the region proposal size

B. The input image is grayscale instead of color

C. The model is missing the classification head

D. The learning rate is too high

Solution

1. Step 1: Understand mask size mismatch error

   This error usually means the predicted mask dimensions don't match the size of the proposed region where the mask should be applied.

2. Step 2: Check other options

   Input image color or missing classification head won't cause mask size mismatch. Learning rate affects training, not mask size.

3. Final Answer:

   The mask output size does not match the region proposal size -> Option A

4. Quick Check:

   Mask size must match proposal size [OK]

Hint: Check mask and proposal sizes match [OK]

Common Mistakes:

• Blaming input image color for mask size error
• Ignoring mask and proposal size relation
• Confusing training hyperparameters with output shape errors

5. You want to improve Mask R-CNN's mask accuracy on small objects in images. Which approach is best?

hard

A. Remove the mask branch and only use bounding boxes

B. Increase the resolution of the feature maps used for mask prediction

C. Use a simpler backbone network to speed up training

D. Reduce the number of region proposals generated by RPN

Solution

1. Step 1: Identify how to improve mask accuracy for small objects

   Small objects need finer details, so increasing feature map resolution helps the mask branch capture small shapes better.

2. Step 2: Evaluate other options

   Reducing proposals may miss objects, simpler backbone reduces feature quality, and removing mask branch loses mask prediction entirely.

3. Final Answer:

   Increase the resolution of the feature maps used for mask prediction -> Option B

4. Quick Check:

   Higher resolution features = better small object masks [OK]

Hint: Higher feature resolution helps small object masks [OK]

Common Mistakes:

• Reducing proposals thinking it improves accuracy
• Using simpler backbone hurting feature quality
• Removing mask branch losing mask output