import numpy as np

subdirs, dirs, files = os.walk('/content/all_images').__next__()
m = len(files)
print(m)

filenames = []
labels = np.zeros((m, 1))

import os
import shutil

images_dir = '/content/all_images'
filenames_counter = 0
labels_counter = -1

for subdir, dirs, files in os.walk(train_dir):
    #print(files)
    for file in files:
        filenames.append(file)
        labels[filenames_counter, 0] = labels_counter
        filenames_counter = filenames_counter + 1
    labels_counter = labels_counter+1

print(len(filenames))
print(labels.shape)