From 73f6396a293a72477b64efc15ee96a7c765f46bb Mon Sep 17 00:00:00 2001 From: Maksymilian Szygenda Date: Mon, 5 Jun 2023 10:36:55 +0200 Subject: [PATCH] Delete 'machine_learning/neuralNetwork.py' --- machine_learning/neuralNetwork.py | 38 ------------------------------- 1 file changed, 38 deletions(-) delete mode 100644 machine_learning/neuralNetwork.py diff --git a/machine_learning/neuralNetwork.py b/machine_learning/neuralNetwork.py deleted file mode 100644 index bc73099..0000000 --- a/machine_learning/neuralNetwork.py +++ /dev/null @@ -1,38 +0,0 @@ -from keras.preprocessing.image import ImageDataGenerator -from keras.models import Sequential -from keras.layers import Conv2D, MaxPooling2D, Flatten, Dense - -train_data_dir = "garbage_photos" - -input_shape = (150, 150, 3) -num_classes = 5 -batch_size = 32 -epochs = 20 - -train_datagen = ImageDataGenerator(rescale=1./255) - -train_generator = train_datagen.flow_from_directory( - train_data_dir, - target_size=(input_shape[0], input_shape[1]), - batch_size=batch_size, - class_mode='categorical' -) - -model = Sequential() -model.add(Conv2D(32, (3, 3), activation='relu', input_shape=input_shape)) -model.add(MaxPooling2D(pool_size=(2, 2))) -model.add(Conv2D(64, (3, 3), activation='relu')) -model.add(MaxPooling2D(pool_size=(2, 2))) -model.add(Conv2D(128, (3, 3), activation='relu')) -model.add(MaxPooling2D(pool_size=(2, 2))) -model.add(Flatten()) -model.add(Dense(128, activation='relu')) -model.add(Dense(num_classes, activation='softmax')) - -model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy']) - -model.fit(train_generator, epochs=epochs) - -classes = train_generator.class_indices - -model.save("neuralModel.h5") \ No newline at end of file