torchvision detection models in PyTorch - Cheat Sheet & Quick Revision

Recall & Review

beginner

What is the main purpose of torchvision detection models?

Torchvision detection models are used to find and locate objects within images. They can tell what objects are present and where they are by drawing boxes around them.

beginner

Name two popular torchvision detection models.

Two popular torchvision detection models are Faster R-CNN and SSD (Single Shot MultiBox Detector). Both are used for detecting objects but differ in speed and accuracy.

beginner

How do you load a pretrained Faster R-CNN model from torchvision?

You can load it using: <br><code>import torchvision.models.detection as detection<br>model = detection.fasterrcnn_resnet50_fpn(pretrained=True)</code><br>This loads a model trained on COCO dataset ready to detect common objects.

intermediate

What kind of output do torchvision detection models produce?

They output a list of dictionaries, each dictionary contains:<br>- boxes: coordinates of detected objects<br>- labels: class IDs of objects<br>- scores: confidence scores for each detection

intermediate

Why is it important to set the model to evaluation mode during inference?

Setting the model to evaluation mode (model.eval()) disables training behaviors like dropout and batch normalization updates, ensuring consistent and correct predictions.

Which torchvision detection model is known for balancing speed and accuracy well?

A VGG16

B ResNet-50

C AlexNet

D Faster R-CNN

Faster R-CNN is a popular detection model that balances speed and accuracy well. ResNet-50, VGG16, and AlexNet are classification models.

What does the 'boxes' output from a detection model represent?

A Coordinates of detected objects

B Class names of objects

C Image pixel values

D Model weights

The 'boxes' output contains the coordinates (usually x_min, y_min, x_max, y_max) of each detected object in the image.

How do you prepare a torchvision detection model for inference?

A Call model.train()

B Reset model weights

C Call model.eval()

D Increase learning rate

Calling model.eval() sets the model to evaluation mode, which is necessary for correct inference.

Which dataset is commonly used to pretrain torchvision detection models?

A COCO

B ImageNet

C MNIST

D CIFAR-10

COCO (Common Objects in Context) is a large dataset used for object detection and is commonly used to pretrain detection models.

What does the 'scores' output from a detection model indicate?

A Training loss

B Confidence level of each detected object

C Size of the image

D Number of objects detected

The 'scores' represent the confidence level that the detected object belongs to the predicted class.

Explain how to use a pretrained torchvision detection model to detect objects in a new image.

Think about the steps from loading the model to getting predictions.

Describe the structure of the output from torchvision detection models and what each part means.

Focus on what the model tells you about each detected object.