Learn practical skills, build real-world projects, and advance your career
Updated 3 years ago
#Image Classification Using CNN
Deep Learning Course Project
I have used fruits dataset for this course project. The task is to classify this images and train a model to predict any previously unseen images which belongs to one of these classes with max accuracy using deep neural network, CNN and ResNets (Here, CNN).
!pip install jovian --upgrade --quiet
!pip install opendatasets --upgrade --quiet#Download, Explore and prepare dataset
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.utils import download_url
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
%matplotlib inline
matplotlib.rcParams['figure.facecolor'] = '#ffffff'##my project name