MENCREATE TRAINING DATA

import numpy as np

import matplotlib.pyplot as plt

import cv2

import os

from google.colab import drive

drive.mount('/content/gdrive')

root_path = '/content/gdrive/My Drive/ImageTesting/'

CATEGORIES = ["dogs","cats"]

    

training_data = []

def create_training_data():

    for category in CATEGORIES:

        path = root_path+category

        os.chdir(path) #path to cat and dog dir

        class_num = CATEGORIES.index(category)

        for img in os.listdir(path):

            print("sedang proses ="+img)

            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])

                print(len(training_data))

            except Exception as e:

                pass

create_training_data()

print(len(training_data))