!pip install jovian --upgrade --quiet
!pip install opendatasets --upgrade --quiet

project_name = 'assignment-project'

import opendatasets as od

# download dataset from kaggle
od.download('https://www.kaggle.com/gpiosenka/100-bird-species')

import os
import cv2
import time
import numpy as np

import torch
import torch.nn as nn
import torch.nn.functional as F

from torch.utils.data import DataLoader, Dataset
from torch.utils.data import RandomSampler

import torchvision.transforms as T
import torchvision.models as models
from torchvision.utils import make_grid
from torchvision.datasets import ImageFolder
from torchvision.transforms import ToTensor

from matplotlib import pyplot as plt

TRAIN_DIR = "./100-bird-species/train/" # contains train images
VALID_DIR = "./100-bird-species/valid/" # contains valid images
TEST_DIR = "./100-bird-species/test/" # contains test images