Jovian
Sign In
Learn practical skills, build real-world projects, and advance your career
import os
import torch
import torchvision
from torch.utils.data import random_split
project_name = "fake-currency-detection"
DATA_DIR = '../input/nepalicurrencynpr/Nepali Currency'
print(os.listdir(DATA_DIR))
classes = os.listdir(DATA_DIR+'/Train')
print(classes)
['Test', 'Train'] ['Rs. 500', 'Rs. 20', 'Rs. 10', 'Rs.5', 'Rs. 100'] 
ten_files = os.listdir(DATA_DIR + "/Train/Rs. 10")
print("No. of training examples for Rs.10 currency:", len(ten_files))
print(ten_files[:5])
No. of training examples for Rs.10 currency: 82 ['1024.jpg', '1017.jpg', '1063.jpg', '1036.jpg', '106.jpg']