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

Intel Image Classification Dataset - Zero To Gans Course Project

I will work with the Intel Image Classification Dataset. It contains images of 6 different categories.

Exploring the Data

Import necessary libraries:

import os
import torch
import torchvision
import tarfile
import torch.nn as nn
import numpy as np
import torch.nn.functional as F
from torchvision.datasets import ImageFolder
from torch.utils.data import DataLoader
import torchvision.transforms as tt
from torch.utils.data import random_split
from torchvision.utils import make_grid
import matplotlib
import matplotlib.pyplot as plt
import seaborn as sns
from google.colab import drive
%matplotlib inline

I have downloaded the dataset to my Google Drive, so we need to connect with Google Drive to acces the data.