From 2bb9f831f29acfab4d0eae43f3c8470dff841163 Mon Sep 17 00:00:00 2001
From: Daria Vodzinska <darvod@st.amu.edu.pl>
Date: Mon, 15 Jun 2020 02:29:35 +0000
Subject: [PATCH] Zaktualizuj 'agent.py'

---
 agent.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/agent.py b/agent.py
index 4db5b90..49b6e64 100755
--- a/agent.py
+++ b/agent.py
@@ -10,6 +10,8 @@ import pdb
 import numpy as np
 from package_location_classifier.classifier import PackageLocationClassifier
 from genetic_route import genetic_trace_route
+from siec import imgSkan, imageClass
+from keras.models import Sequential
 
 class Node:
 	def __init__(self, coord_x, coord_y, package=None, is_rack=False):
@@ -205,7 +207,13 @@ class Agent:
 				break
 		if dst_package is not None:
 			return dst_package
+	
+    nmodel = Sequential()
+	imageClass(nmodel)
+	nmodel.load_weights('model_weights.h5')
 
+	imgSkan()
+    
 	def rack_heuristics(self, start, goal, can_place):
 		heur_can_place = not can_place
 		diff_x = pow(goal.x - start.x, 2)