Jovian
Sign In
Learn practical skills, build real-world projects, and advance your career
3 years ago
import cv2
from keras.preprocessing.image import ImageDataGenerator
import glob
from keras.models import Sequential
from keras.layers import Activation, Dropout, Flatten, Dense, Conv2D, MaxPool2D
import matplotlib.pyplot as plt
import os
Using TensorFlow backend.

Getting first 20 Folder Names

fldrs = glob.glob('train-images/*')
flds = []
for i in fldrs[:20]:
    flds.append(i.split('\\')[1])
    
print(flds)
['001.Black_footed_Albatross', '002.Laysan_Albatross', '003.Sooty_Albatross', '004.Groove_billed_Ani', '005.Crested_Auklet', '006.Least_Auklet', '007.Parakeet_Auklet', '008.Rhinoceros_Auklet', '009.Brewer_Blackbird', '010.Red_winged_Blackbird', '011.Rusty_Blackbird', '012.Yellow_headed_Blackbird', '013.Bobolink', '014.Indigo_Bunting', '015.Lazuli_Bunting', '016.Painted_Bunting', '017.Cardinal', '018.Spotted_Catbird', '019.Gray_Catbird', '020.Yellow_breasted_Chat']

Data Cleaning and Preparation

  • Testing set preparation
folders = glob.glob('images/*')