HomeCoursesGetting Started with Image Classification with PyTorch

Beginner

Getting Started with Image Classification with PyTorch
Save for later

Gain insights into image classification with PyTorch. Learn about data preprocessing, model training, fine-tuning, and deploying models using ONNX for real-world applications.

Join 2.6 million developers at

Table of Contents

Learner Reviews

Course Overview

PyTorch is a machine learning framework used in a wide array of popular applications, including Tesla’s Autopilot and Pyro, Uber’s probabilistic modeling engine. This course is an introduction to image classification using PyTorch’s computer vision models for training and tuning your own model. You’ll start with the fundamental concepts of applying machine learning and its applications to image classification before exploring the process of training your AI model. You’ll prepare data for intake by the comp...Show More

PyTorch is a machine learning framework used in a wide array of popular applications, including Tesla’s Autopilot and Pyro, Uber...Show More

WHAT YOU'LL LEARN

A basic overview of the PyTorch Image Model

The ability to fine-tune custom image classification models

A working knowledge of deploying models as REST API

A familiarity with converting PyTorch models into ONNX format

A basic overview of the PyTorch Image Model

TAKEAWAY SKILLS

Database

Python

SQL

Course Content

50 Lessons6 Quizzes

Introduction

4 Lessons

Get familiar with image classification, techniques, metrics, and PyTorch Image Model framework.

Overview of Image Classification

Image Classification Techniques

Image Classification Metrics

PyTorch Image Model Framework

Basic Concepts

8 Lessons

Look at essential PyTorch image classification, including models, datasets, preprocessing, and inference.

Create Model

List and Search Supported Models

Dataset Format

Load and Preprocess Images

Prediction Using Pre-trained Resnet50

Prediction Using Pre-trained EfficientNet

Quiz (Basic)

Exercise (Basic)

Augmentation

7 Lessons

Examine augmentation techniques to diversify datasets, improve model performance, and mitigate overfitting.

Training a Neural Network with Augmentation

Random Resized Crop and Interpolation

Quiz (Augmentation)

Loss

4 Lessons

Grasp the fundamentals of loss functions to improve model accuracy in PyTorch.

Loss Functions

Asymmetric Loss

Jensen-Shannon Divergence and Cross-Entropy Loss

Quiz (Loss)

Training

7 Lessons

Solve problems in image classification training using PyTorch, models, and techniques.

Training Workflow

ResNet50

EfficientNet

Fine-tuning Custom Model

Training with an EMA (Exponential Moving Average)

Quiz (Training)

Exercise (Training)

Model Conversion

9 Lessons

Follow the process of converting and serving models across PyTorch, ONNX, TensorFlow, and TFLite.

Serving a Pytorch Model

Convert PyTorch Model into an ONNX Model

Serving an ONNX Model

Convert ONNX into a Tensorflow Model

Serving a TensorFlow Model

Convert a TensorFlow Model into a TFLite Model

Serving a TFLite Model

Quiz (Model Conversion)

Exercise (Model Conversion)

Deployment

5 Lessons

Practice using FastAPI to deploy image classification models with HTTP methods and REST API integration.

Exercise (Deployment)

Appendix

5 Lessons

Learn how to use virtual environments, Python packages, training arguments, and deployment dependencies.

Deployment Dependencies

Conclusion

1 Lessons

Solve challenges with essential PyTorch image classification concepts, model deployment, and performance improvement.

Ending Note

Certificate of Completion

Showcase your accomplishment by sharing your certificate of completion.

Course Author:

Ng Wai Foong

Join 2.6 million learners and start transforming your career today

Trusted by 2.6 million developers working at companies

"These are high-quality courses. Trust me. I own around 10 and the price is worth it for the content quality. EducativeInc came at the right time in my career. I'm understanding topics better than with any book or online video tutorial I've done. Truly made for developers. Thanks"

Anthony Walker

@_webarchitect_

"Just finished my first full #ML course: Machine learning for Software Engineers from Educative, Inc. ... Highly recommend!"

Evan Dunbar

ML Engineer

"You guys are the gold standard of crash-courses... Narrow enough that it doesn't need years of study or a full blown book to get the gist, but broad enough that an afternoon of Googling doesn't cut it."

Carlos Matias La Borde

Software Developer

"I spend my days and nights on Educative. It is indispensable. It is such a unique and reader-friendly site"

Souvik Kundu

Front-end Developer

"Your courses are simply awesome, the depth they go into and the breadth of coverage is so good that I don't have to refer to 10 different websites looking for interview topics and content."

Vinay Krishnaiah

Software Developer

"I've tried probably 5-7 different sites and Educative is easily the best. It perfectly blends explanation with interactivity"

Eric Downs

Musician/Entrepeneur

Hands-on Learning Powered by AI

See how Educative uses AI to make your learning more immersive than ever before.

Instant Code Feedback

Evaluate and debug your code with the click of a button. Get real-time feedback on test cases, including time and space complexity of your solutions.

AI-Powered Mock Interviews

Put your skills to the test in a simulated interview setting. Receive personalized feedback based on your performance. Available in Premium & Premium Plus plans.

Adaptive Learning

At various checkpoints throughout Educative courses, you will be prompted to take a quick assessment. Receive a condensed curriculum tailored to your strengths and skill gaps.

Explain with AI

Select any text within any Educative course, and get an instant explanation — without ever leaving your browser.

AI Code Mentor

AI Code Mentor helps you quickly identify errors in your code, learn from your mistakes, and nudge you in the right direction — just like a 1:1 tutor!

Free Resources

FOR TEAMS

Interested in this course for your business or team?

Unlock this course (and 1,000+ more) for your entire org with DevPath

Getting Started with Image Classification with PyTorch Save for later

Getting Started with Image Classification with PyTorch
Save for later