Jovian
Sign In
Learn practical skills, build real-world projects, and advance your career
3 years ago
!pip install jovian --upgrade --quiet
!pip install opendatasets --upgrade --quiet
import opendatasets as od
import os
import torch
import torchvision
import torch.nn as nn
import torch.nn.functional as F
from torchvision.datasets import ImageFolder
from torch.utils.data import DataLoader
import torchvision.transforms as T
from torch.utils.data import random_split
import torch.utils.data as data
from torchvision.utils import make_grid
import torch.optim as optim

import matplotlib
import matplotlib.pyplot as plt
%matplotlib inline

matplotlib.rcParams['figure.facecolor'] = '#ffffff'
dataset_url="https://www.kaggle.com/meowmeowmeowmeowmeow/gtsrb-german-traffic-sign"
od.download(dataset_url)
0%| | 0.00/612M [00:00<?, ?B/s]
Downloading gtsrb-german-traffic-sign.zip to ./gtsrb-german-traffic-sign
100%|██████████| 612M/612M [00:14<00:00, 45.6MB/s] 
input_data="/content/gtsrb-german-traffic-sign/Train"
c1=os.listdir(input_data)
print(c1)
['5', '10', '23', '35', '22', '6', '36', '24', '18', '29', '42', '26', '41', '33', '2', '37', '3', '0', '32', '30', '25', '13', '14', '28', '31', '1', '16', '40', '15', '38', '17', '8', '34', '27', '20', '7', '12', '21', '9', '39', '11', '4', '19'] 
train_dataset="/content/gtsrb-german-traffic-sign/Train"
classes=os.listdir(train_dataset)
print(classes)
['5', '10', '23', '35', '22', '6', '36', '24', '18', '29', '42', '26', '41', '33', '2', '37', '3', '0', '32', '30', '25', '13', '14', '28', '31', '1', '16', '40', '15', '38', '17', '8', '34', '27', '20', '7', '12', '21', '9', '39', '11', '4', '19']