Learn practical skills, build real-world projects, and advance your career
Updated 3 years ago
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.