pip install opendatasets --upgrade --quiet

import os
import opendatasets as od
import matplotlib.pyplot as plt
from collections import defaultdict 

import torch
import torch.nn as nn
from torchvision import transforms
from torch.utils.data import DataLoader
from torchvision.utils import make_grid
from torchvision.datasets import ImageFolder

dataset_url = 'https://www.kaggle.com/moltean/fruits'
od.download(dataset_url)

data_dir = '/content/fruits/fruits-360'
print('Fruits-360 directory:', ', '.join(os.listdir(data_dir)))

train_dataset = ImageFolder(
    root = data_dir + '/Training',
    transform = transforms.ToTensor())

valid_dataset = ImageFolder(
    root = data_dir + '/Test',
    transform = transforms.ToTensor())

print('Train Images: ', len(train_dataset))
print('Test Images : ', len(valid_dataset))