import numpy as np
import matplotlib.pyplot as plt
import cv2
import os
DATADIR=“D:/ML/DATA/kagglecatsanddogs_3367a/PetImages”
CATEGORIES=[“Dog”,“Cat”]
IMG_SIZE=50
training_data=[]
def create_training_data():
for category in CATEGORIES:
path=os.path.join(DATADIR,category)
class_num=CATEGORIES.index(category)
for img in os.listdir(path):
try:
img_array=cv2.imread(os.path.join(path,img),cv2.IMREAD_GRAYSCALE)
new_array=cv2.resize(img_array,(IMG_SIZE,IMG_SIZE))
training_data.append([new_array,class_num])
except Exception as e:
pass
create_training_data()
random.shuffle(training_data)
import random
for sample in training_data[:10]:
print(sample[1])
x=[]
y=[]
for features,label in training_data:
x.append(features)
y.append(label)
x=np.array(x).reshape(-1,IMG_SIZE,IMG_SIZE,1)
import pickle
pickle_out=open(“x.pickle”,“wb”)
pickle.dump(x,pickle_out)
pickle_out.close()
pickle_out=open(“y.pickle”,“wb”)
pickle.dump(x,pickle_out)
pickle_out.close()
pickle_in=open(“x.pickle”,“rb”)
x=pickle.load(pickle_in)