Classifying Natural Scenes using Transfer Learning

import os
import torch
import torchvision
from torch.utils.data import random_split
from torchvision.datasets import ImageFolder
import torchvision.transforms as tt
from torchvision.transforms import ToTensor
from torchvision.transforms import Resize
from torchvision.transforms import Compose
import torch.nn as nn
import torch.nn.functional as F
import torchvision.models as models
from tqdm.notebook import tqdm

project_name = "zero-to-gans-course-project"

data_dir = "../input/intel-image-classification"

train_dir = data_dir + "/seg_train/seg_train"
test_dir = data_dir + "/seg_test/seg_test"

classes = ["buildings", "forest", "glacier", "mountain", "sea", "street"]
print("The different classes in the dataset:",classes)