YOLO concept in PyTorch - Practice Problems & Coding Challenges

Challenge - 5 Problems

🧠 Conceptual

intermediate

What is the main advantage of YOLO compared to traditional object detection methods?

YOLO (You Only Look Once) is a popular object detection model. What is its main advantage compared to older methods like sliding window or region proposal based detectors?

A It processes the entire image in a single pass, making detection faster.

B It uses multiple passes over the image to improve accuracy.

C It requires manual feature engineering for each object class.

D It detects objects by scanning small patches independently.

💡 Hint

Think about how YOLO treats the image differently from methods that look at many small parts separately.

YOLO divides the image into a grid and predicts bounding boxes and class probabilities for all grid cells in one forward pass, making it very fast compared to methods that scan many regions separately.

❓ Predict Output

intermediate

What is the output shape of YOLO's final layer for a 7x7 grid with 2 boxes per cell and 20 classes?

YOLO divides the image into an SxS grid. Each grid cell predicts B bounding boxes and C class probabilities. Given S=7, B=2, and C=20, what is the shape of the output tensor from the final layer?

A [7, 7, 30]

B [7, 7, 14]

C [7, 7, 24]

D [7, 7, 28]

💡 Hint

Each bounding box predicts 5 values (x, y, w, h, confidence) plus class probabilities per cell.

Each cell predicts B=2 boxes, each with 5 values, so 2*5=10 values, plus C=20 class probabilities, total 30 values per cell. So output shape is [7,7,30].

❓ Model Choice

advanced

Which PyTorch layer is best suited to implement YOLO's final prediction layer?

YOLO's final layer outputs a tensor with shape [batch_size, S, S, B*5 + C]. Which PyTorch layer is most appropriate to produce this output from a feature map?

A nn.Conv2d with kernel size 3 and output channels = B*5 + C

B nn.Conv2d with kernel size 1 and output channels = B*5 + C

C nn.Linear with output features = S*S*(B*5 + C)

D nn.MaxPool2d with kernel size 2

💡 Hint

The layer should keep spatial dimensions but change channel depth to prediction size.

A 1x1 convolution changes the number of channels per spatial location without changing height and width, perfect for predicting bounding boxes and classes per grid cell.

❓ Metrics

advanced

Which metric best evaluates YOLO's object detection performance?

YOLO predicts bounding boxes and class labels. Which metric is most appropriate to evaluate its detection accuracy?

A Accuracy

B Mean Squared Error (MSE)

C Mean Average Precision (mAP)

D Cross-Entropy Loss

💡 Hint

Consider a metric that accounts for both localization and classification correctness.

mAP measures how well predicted boxes match ground truth boxes and correct class labels, combining localization and classification performance.

🔧 Debug

expert

Why does YOLO sometimes miss small objects in images?

YOLO divides the image into a grid and predicts boxes per cell. Why might it struggle to detect very small objects?

A YOLO only predicts one bounding box per image.

B YOLO uses too many grid cells, causing overfitting on small objects.

C YOLO's loss function ignores small objects during training.

D Small objects may fall into one grid cell and be overshadowed by larger objects, causing missed detections.

💡 Hint

Think about how grid size affects the ability to localize small objects.

Because YOLO predicts a fixed number of boxes per grid cell, small objects that occupy less than a cell can be missed or confused with larger objects in the same cell.

Practice

1. What is the main advantage of the YOLO (You Only Look Once) model in object detection?

easy

A. It uses a very slow but highly accurate method.

B. It detects multiple objects in an image quickly in a single pass.

C. It only detects one object per image.

D. It requires multiple passes over the image to detect objects.

Solution

1. Step 1: Understand YOLO's detection method

   YOLO processes the entire image once to detect all objects, unlike methods that scan multiple times.

2. Step 2: Compare options with YOLO's feature

   It detects multiple objects in an image quickly in a single pass correctly states YOLO's speed and single-pass detection. Other options contradict this.

3. Final Answer:

   It detects multiple objects in an image quickly in a single pass. -> Option B

4. Quick Check:

   YOLO speed = single pass detection [OK]

Hint: YOLO means one pass for all objects [OK]

Common Mistakes:

• Thinking YOLO scans image multiple times
• Believing YOLO detects only one object
• Confusing YOLO with slow detection methods

2. Which of the following is the correct way to load a pretrained YOLO model in PyTorch?

easy

A. model = torch.hub.download('yolov5s')

B. model = torch.load('yolov5s.pt')

C. model = torch.yolo.load('yolov5s')

D. model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)

Solution

1. Step 1: Recall PyTorch hub usage for YOLO

   PyTorch hub uses torch.hub.load with repo and model name to load pretrained models.

2. Step 2: Check each option

   model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True) uses correct torch.hub.load syntax with pretrained=True. model = torch.load('yolov5s.pt') loads a file but not via hub. model = torch.yolo.load('yolov5s') and D use invalid methods.

3. Final Answer:

   model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True) -> Option D

4. Quick Check:

   PyTorch hub load syntax = model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True) [OK]

Hint: Use torch.hub.load with repo and model name [OK]

Common Mistakes:

• Using torch.load without specifying hub
• Trying non-existent torch.yolo module
• Confusing download with load

3. Given this PyTorch code snippet using YOLO model:

import torch
model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True)
img = torch.zeros((1, 3, 640, 640))
pred = model(img)
print(len(pred.xyxy[0]))

What does the printed number represent?

medium

A. The number of image channels

B. The batch size of the input tensor

C. The number of detected objects in the image

D. The number of model layers

Solution

1. Step 1: Understand pred.xyxy output

   pred.xyxy[0] contains bounding boxes for detected objects in the first image of the batch.

2. Step 2: len(pred.xyxy[0]) counts detected boxes

   Counting these boxes gives the number of objects detected in the image.

3. Final Answer:

   The number of detected objects in the image -> Option C

4. Quick Check:

   Length of pred.xyxy[0] = detected objects count [OK]

Hint: Length of pred.xyxy[0] = detected objects [OK]

Common Mistakes:

• Confusing output length with image channels
• Thinking it shows batch size
• Assuming it counts model layers

4. You run YOLO detection code but get an error: AttributeError: 'NoneType' object has no attribute 'xyxy'. What is the most likely cause?

medium

A. The model was not loaded correctly, so prediction returned None.

B. The input image tensor has wrong shape.

C. YOLO does not support batch size 1.

D. The torch library is outdated.

Solution

1. Step 1: Analyze the error message

   'NoneType' means prediction output is None, so accessing .xyxy fails.

2. Step 2: Identify cause of None output

   If model loading fails or is incorrect, model(img) returns None, causing this error.

3. Final Answer:

   The model was not loaded correctly, so prediction returned None. -> Option A

4. Quick Check:

   None output means model load failed [OK]

Hint: None output means model load failed [OK]

Common Mistakes:

• Blaming input shape without checking model load
• Assuming batch size 1 is unsupported
• Ignoring error message details

5. You want to detect only 'person' objects using a pretrained YOLO model in PyTorch. Which approach correctly filters the detections?

hard

A. Filter pred.xyxy[0] by checking if pred.names[class_id] == 'person' for each detected box.

B. Set model to detect only 'person' by passing classes=['person'] when loading.

C. Use a separate model trained only on persons instead of YOLO.

D. Manually crop image to only person areas before detection.

Solution

1. Step 1: Understand YOLO output structure

   YOLO outputs boxes with class IDs; pred.names maps IDs to labels like 'person'.

2. Step 2: Filter detections by class label

   Filtering pred.xyxy[0] by checking if class label equals 'person' keeps only person detections.

3. Final Answer:

   Filter pred.xyxy[0] by checking if pred.names[class_id] == 'person' for each detected box. -> Option A

4. Quick Check:

   Filter detections by class label = Filter pred.xyxy[0] by checking if pred.names[class_id] == 'person' for each detected box. [OK]

Hint: Filter detections by class label after prediction [OK]

Common Mistakes:

• Trying to limit classes at model load (not supported directly)
• Using separate model unnecessarily
• Cropping image instead of filtering results