Cutout and CutMix in Computer Vision - Model Pipeline Trace

Model Pipeline - Cutout and CutMix

This pipeline shows how Cutout and CutMix help improve image classification models by changing training images to make the model learn better. Cutout hides parts of images, and CutMix mixes parts of two images.

Data Flow - 4 Stages

1Original Dataset

5000 images x 32 x 32 x 3→Raw images with labels→5000 images x 32 x 32 x 3

Image of a cat with label 'cat'

↓

2Cutout Augmentation

5000 images x 32 x 32 x 3→Randomly mask a square patch in each image→5000 images x 32 x 32 x 3

Cat image with an 8x8 black square hiding part of the cat

↓

3CutMix Augmentation

5000 images x 32 x 32 x 3→Mix patches from two images and combine labels proportionally→5000 images x 32 x 32 x 3

Image with half cat and half dog, label is 0.6 cat + 0.4 dog

↓

4Model Training

Batch of 128 images x 32 x 32 x 3→Train CNN on augmented images and labels→Model weights updated

CNN learns features from images with cutout and cutmix

Training Trace - Epoch by Epoch

Loss
1.2 |*       
1.0 | *      
0.8 |  *     
0.6 |   *    
0.4 |    *   
0.2 |     *  
0.0 +--------
     1 5 10 15 20 Epochs

Epoch	Loss ↓	Accuracy ↑	Observation
1	1.2	0.45	Model starts learning, loss high, accuracy low
5	0.8	0.65	Loss decreases, accuracy improves with augmentation
10	0.5	0.80	Model learns robust features, accuracy rises
15	0.35	0.88	Loss low, accuracy high, good generalization
20	0.30	0.90	Training converged with strong performance

Prediction Trace - 6 Layers

Layer 1: Input Image

Layer 2: Convolutional Layers

Layer 3: Pooling Layers

Layer 4: Fully Connected Layers

Layer 5: Softmax Activation

Layer 6: Final Prediction

Model Quiz - 3 Questions

What does the Cutout augmentation do to the training images?

A Changes image colors randomly

B Mixes two images together

C Randomly hides a square patch in the image

D Rotates the image by 90 degrees

Key Insight

Cutout and CutMix help the model learn better by making training images more varied and challenging. This leads to lower loss and higher accuracy, showing the model generalizes well to new images.

Practice

1. What is the main purpose of the Cutout technique in computer vision?

easy

A. To remove a patch from an image to improve model robustness

B. To combine two images by overlaying one on the other

C. To increase image resolution for better detail

D. To convert images to grayscale for simpler processing

Solution

1. Step 1: Understand Cutout's function

   Cutout removes a small patch from an image during training to force the model to learn from other parts.

2. Step 2: Compare options

   Only To remove a patch from an image to improve model robustness describes removing a patch to improve robustness. Others describe different techniques.

3. Final Answer:

   To remove a patch from an image to improve model robustness -> Option A

4. Quick Check:

   Cutout = patch removal [OK]

Hint: Cutout = cut a patch out of the image [OK]

Common Mistakes:

• Confusing Cutout with CutMix
• Thinking Cutout increases resolution
• Assuming Cutout converts image color

2. Which of the following is the correct way to mix labels in CutMix augmentation in Python using PyTorch?

easy

A. new_label = label1 * lambda + label2 * (1 - lambda)

B. new_label = label1 + label2

C. new_label = label1 - label2

D. new_label = label1 / label2

Solution

1. Step 1: Recall CutMix formula

   CutMix creates a new label by mixing two labels weighted by lambda and (1 - lambda).

2. Step 2: Match formula to options

   new_label = label1 * lambda + label2 * (1 - lambda) correctly shows weighted sum of two labels. Others are incorrect operations.

3. Final Answer:

   new_label = label1 * lambda + label2 * (1 - lambda) -> Option A

4. Quick Check:

   CutMix = weighted sum [OK]

Hint: CutMix blends labels weighted by lambda [OK]

Common Mistakes:

• Using addition without weights
• Subtracting or dividing labels incorrectly
• Confusing CutMix with Cutout

3. Given the following code snippet applying Cutout, what will be the shape of the output image tensor?

import torch
import torchvision.transforms as T

cutout = T.RandomErasing(p=1.0, scale=(0.1, 0.1))
image = torch.randn(3, 224, 224)
output = cutout(image)
print(output.shape)

medium

A. torch.Size([3, 112, 112])

B. torch.Size([1, 224, 224])

C. torch.Size([224, 224, 3])

D. torch.Size([3, 224, 224])

Solution

1. Step 1: Understand RandomErasing effect

   RandomErasing (Cutout) masks part of the image but does not change its shape.

2. Step 2: Check input and output shapes

   Input is (3, 224, 224), output remains same shape after Cutout.

3. Final Answer:

   torch.Size([3, 224, 224]) -> Option D

4. Quick Check:

   Cutout keeps shape same [OK]

Hint: Cutout masks but keeps image size [OK]

Common Mistakes:

• Thinking Cutout changes image dimensions
• Confusing channel order with shape
• Assuming output is batch dimension added

4. The following code attempts to apply CutMix but causes an error. What is the main issue?

def cutmix(image1, image2, lam):
    bbx1, bby1, bbx2, bby2 = rand_bbox(image1.size(), lam)
    image1[:, bbx1:bbx2, bby1:bby2] = image2[:, bbx1:bbx2, bby1:bby2]
    return image1

# image1 and image2 are tensors of shape (3, 224, 448)
# lam is a float between 0 and 1

medium

A. rand_bbox returns coordinates in (x, y) but slicing expects (height, width)

B. rand_bbox expects size as tuple (H, W) instead of tensor.size() (C, H, W)

C. The function modifies image1 in-place which is not allowed

D. The function does not return the mixed label

Solution

1. Step 1: Analyze rand_bbox input and slicing

   rand_bbox expects size as (height, width), but image1.size() returns (channels, height, width).

2. Step 2: Identify mismatch in size argument

   Passing image1.size() (C, H, W) instead of (H, W) causes rand_bbox to compute wrong coordinates, leading to slicing errors.

3. Final Answer:

   rand_bbox expects size as tuple (H, W) instead of tensor.size() (C, H, W) -> Option B

4. Quick Check:

   Size argument mismatch = error [OK]

Hint: Pass (H, W) size to rand_bbox, not (C, H, W) [OK]

Common Mistakes:

• Ignoring coordinate order difference
• Assuming in-place modification causes error
• Forgetting to return mixed label

5. You want to improve your image classifier's robustness by combining Cutout and CutMix. Which approach correctly applies both augmentations during training?

hard

A. Apply Cutout to each image separately, then combine them with CutMix

B. Apply Cutout only after training with CutMix is complete

C. Apply CutMix first to combine images, then apply Cutout to mask patches on the mixed image

D. Apply CutMix only on validation data and Cutout only on training data

Solution

1. Step 1: Understand augmentation order

   CutMix combines two images, then Cutout can mask parts of the combined image to increase robustness.

2. Step 2: Evaluate options for correct sequence

   Apply CutMix first to combine images, then apply Cutout to mask patches on the mixed image applies CutMix first, then Cutout, which is the correct order to augment the mixed image effectively.

3. Final Answer:

   Apply CutMix first to combine images, then apply Cutout to mask patches on the mixed image -> Option C

4. Quick Check:

   CutMix then Cutout = best practice [OK]

Hint: Mix images first, then mask patches [OK]

Common Mistakes:

• Applying Cutout before mixing images
• Using augmentations only on validation data
• Applying augmentations after training