Face Mask Detection - Notebook by Ishan Chugh (ishaanchugh06)

Learn practical skills, build real-world projects, and advance your career

Updated 3 years ago

Run on Colab

Run on Kaggle

Run on Binder

Duplicate

!pip install jovian --upgrade --quiet

Face Mask Detection Project using PyTorch

TODO

Download the dataset
Import the dataset using PyTorch
Explore the dataset
Prepare the dataset for training

import torch
import torchvision
import torch.nn as nn
from tqdm.notebook import tqdm
import torch.nn.functional as F
import matplotlib.pyplot as plt
import torchvision.transforms as T
import torchvision.models as models
from torch.utils.data import DataLoader
from torchvision.utils import make_grid
from torchvision.datasets import ImageFolder

Download the dataset

I found this dataset on Kaggle and I am using opendatasets library to download this dataset.