s473616/sztuczna_inteligencja_2023_smieciarka
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

neuron network model

Browse Source
This commit is contained in:
Pawel Felcyn 2023-06-17 18:33:23 +02:00
parent c267515ee1
commit 2d7ed16185
2 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
machine_learning/neuralModel.h5
View File

Binary file not shown.

3
machine_learning/neuralNetwork.py
View File

@ -1,8 +1,11 @@
import os
from keras.preprocessing.image import ImageDataGenerator from keras.preprocessing.image import ImageDataGenerator
from keras.models import Sequential from keras.models import Sequential
from keras.layers import Conv2D, MaxPooling2D, Flatten, Dense from keras.layers import Conv2D, MaxPooling2D, Flatten, Dense
train_data_dir = "garbage_photos" train_data_dir = "garbage_photos"
location = os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file__)))
train_data_dir = os.path.join(location, train_data_dir)
input_shape = (150, 150, 3) input_shape = (150, 150, 3)
num_classes = 5 num_classes = 5