Skip to Content
Machine Learning with PyTorch and Scikit-Learn
book

Machine Learning with PyTorch and Scikit-Learn

by Sebastian Raschka, Yuxi (Hayden) Liu, Vahid Mirjalili
February 2022
Intermediate to advanced
774 pages
21h 56m
English
Packt Publishing
Content preview from Machine Learning with PyTorch and Scikit-Learn

14

Classifying Images with Deep Convolutional Neural Networks

In the previous chapter, we looked in depth at different aspects of the PyTorch neural network and automatic differentiation modules, you became familiar with tensors and decorating functions, and you learned how to work with torch.nn. In this chapter, you will now learn about convolutional neural networks (CNNs) for image classification. We will start by discussing the basic building blocks of CNNs, using a bottom-up approach. Then, we will take a deeper dive into the CNN architecture and explore how to implement CNNs in PyTorch. In this chapter, we will cover the following topics:

  • Convolution operations in one and two dimensions
  • The building blocks of CNN architectures
  • Implementing ...
Become an O’Reilly member and get unlimited access to this title plus top books and audiobooks from O’Reilly and nearly 200 top publishers, thousands of courses curated by job role, 150+ live events each month,
and much more.
Start your free trial

You might also like

Hands-On Machine Learning with Scikit-Learn and PyTorch

Hands-On Machine Learning with Scikit-Learn and PyTorch

Aurélien Géron

Publisher Resources

ISBN: 9781801819312Supplemental Content