2023-06-19 11:52:44 +02:00
2 changed files with 3 additions and 0 deletions
--- a/machine_learning/neuralModel.h5
+++ b/machine_learning/neuralModel.h5
--- a/machine_learning/neuralNetwork.py
+++ b/machine_learning/neuralNetwork.py
@ -1,8 +1,11 @@
+import os
 from keras.preprocessing.image import ImageDataGenerator
 from keras.models import Sequential
 from keras.layers import Conv2D, MaxPooling2D, Flatten, Dense

 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)
 num_classes = 5