big image database update

saved model import

ForkliftAgent.py

@@ -1,5 +1,7 @@
 from typing import Tuple, List
 
+from tensorflow import keras
+
 from AgentBase import AgentBase
 from PatchAgent import PatchAgent
 from PatchType import PatchType
@@ -37,6 +39,9 @@ class ForkliftAgent(AgentBase):
     def queue_movement_actions(self, movement_actions: List[Action]):
         self.action_queue.extend(movement_actions)
 
+    imageModel = keras.models.load_model('imageClasification/my_model.h5')
+    print("imageModel loaded")
+
     def move(self):
         if len(self.action_queue) > 0:
             action = self.action_queue.pop(0)

imageClasification/imageResize.py

@@ -1,11 +0,0 @@
-from PIL import Image
-import os
-import PIL
-import glob
-
-
-images = [file for file in os.listdir() if file.endswith(('jpeg', 'png', 'jpg'))]
-for image in images:
-    img = Image.open(image)
-    img.thumbnail((256,256))
-    img.save(image, optimize=True, quality=60)