Reader small image

You're reading from  Hands-On Computer Vision with Detectron2

Product typeBook
Published inApr 2023
Reading LevelBeginner
PublisherPackt
ISBN-139781800561625
Edition1st Edition
Languages
Tools
Right arrow
Author (1)
Van Vung Pham
Van Vung Pham
author image
Van Vung Pham

Van Vung Pham is a passionate research scientist in machine learning, deep learning, data science, and data visualization. He has years of experience and numerous publications in these areas. He is currently working on projects that use deep learning to predict road damage from pictures or videos taken from roads. One of the projects uses Detectron2 and Faster R-CNN to predict and classify road damage and achieve state-of-the-art results for this task. Dr. Pham obtained his PhD from the Computer Science Department, at Texas Tech University, Lubbock, Texas, USA. He is currently an assistant professor at the Computer Science Department, Sam Houston State University, Huntsville, Texas, USA.
Read more about Van Vung Pham

Right arrow

The backbone network

Chapter 2 discusses the typical backbone networks for Detectron2. They include ResNet50, ResNet101, ResNeXt101, and their variants. This section inspects the ResNet50 architecture as an example. However, the idea remains the same for other base models (backbone networks). Figure 4.3 summarizes the steps to inspect the backbone network. Specifically, we pass a tensor of data for a single image to the backbone, and the backbone (ResNet50, in this case) gives out a tensor. This output tensor is the extracted salient feature of the input image.

Figure 4.3: The backbone network

Figure 4.3: The backbone network

Specifically, from the default Detectron2’s predictor, we can access the backbone network using the following code snippet:

backbone = predictor.model.backbone
type(backbone)

This code snippet should print out the following:

detectron2.modeling.backbone.resnet.ResNet

The following code snippet reveals the backbone’s architecture:

print...
lock icon
The rest of the page is locked
Previous PageNext Page
You have been reading a chapter from
Hands-On Computer Vision with Detectron2
Published in: Apr 2023Publisher: PacktISBN-13: 9781800561625

Author (1)

author image
Van Vung Pham

Van Vung Pham is a passionate research scientist in machine learning, deep learning, data science, and data visualization. He has years of experience and numerous publications in these areas. He is currently working on projects that use deep learning to predict road damage from pictures or videos taken from roads. One of the projects uses Detectron2 and Faster R-CNN to predict and classify road damage and achieve state-of-the-art results for this task. Dr. Pham obtained his PhD from the Computer Science Department, at Texas Tech University, Lubbock, Texas, USA. He is currently an assistant professor at the Computer Science Department, Sam Houston State University, Huntsville, Texas, USA.
Read more about Van Vung Pham