refactor

src/find.py

@@ -11,9 +11,9 @@ from PIL import Image
 import os
 
 
-for filename in os.listdir('./resources/smieci/plastic'):
+for filename in os.listdir('../resources/smieci/plastic'):
 
-    img = Image.open('./resources/smieci/plastic/'+filename)
+    img = Image.open('../resources/smieci/plastic/'+filename)
     pil_to_tensor = transforms.ToTensor()(img).unsqueeze_(0)
     if(str(pil_to_tensor.shape[1]) == '1'):
         print(filename)

src/game.py

@@ -10,18 +10,18 @@ import uczenie_adamB as adamB
 
 
 smieci_path = ''
-smieci_w_kontenerach = "resources\\smieci w kontenerach"
+smieci_w_kontenerach = "..\\resources\\smieci w kontenerach"
 
 # PODAJ OSOBE PRZED URUCHOMIENIEM (kacper/adamB/adamO)
 osoba = 'kacper'
 rfc = None
 
 if osoba == 'kacper':
-    smieci_path = 'resources\\smieci'
+    smieci_path = '..\\resources\\smieci'
 elif osoba == 'adamB':
-    smieci_path = 'resources\\smieci'
+    smieci_path = '..\\resources\\smieci'
 else:
-    smieci_path = 'resources\\smieci_stare'
+    smieci_path = '..\\resources\\smieci_stare'
     rfc = adamO.rozpocznijUczenie()
 
 pygame.init()
@@ -165,9 +165,9 @@ def rysowaniePlanszy(obiekty):
                               HEIGHT])
 
     obiekty["obraz"].blit(pygame.image.load(
-        "resources/plansza/wysypisko.jpg"), (5, 5))
+        "../resources/plansza/wysypisko.jpg"), (5, 5))
     obiekty["obraz"].blit(pygame.image.load(
-        "resources/plansza/jezioro.png"), (395, 655))
+        "../resources/plansza/jezioro.png"), (395, 655))
     obiekty["plansza"][6, 10].setJestPrzeszkoda(True)
     obiekty["plansza"][6, 11].setJestPrzeszkoda(True)
     obiekty["plansza"][7, 10].setJestPrzeszkoda(True)
@@ -216,32 +216,32 @@ def utworzObiekty():
 
     kontener_szklo = modele.Kontener(4, 4, "glass")
     kontener_szklo.setImage(pygame.image.load(
-        "resources/plansza/pojemnik_szklo.png"))
+        "../resources/plansza/pojemnik_szklo.png"))
     plansza[4, 4].setJestKontenerem(True)
     plansza[4, 4].setObiekt(kontener_szklo)
 
     kontener_metal = modele.Kontener(0, 4, "metal")
     kontener_metal.setImage(pygame.image.load(
-        "resources/plansza/pojemnik_metal.png"))
+        "../resources/plansza/pojemnik_metal.png"))
     plansza[0, 4].setJestKontenerem(True)
     plansza[0, 4].setObiekt(kontener_metal)
 
     kontener_papier = modele.Kontener(4, 0, "paper")
     kontener_papier.setImage(pygame.image.load(
-        "resources/plansza/pojemnik_papier.png"))
+        "../resources/plansza/pojemnik_papier.png"))
     plansza[4, 0].setJestKontenerem(True)
     plansza[4, 0].setObiekt(kontener_papier)
 
     kontener_plastik = modele.Kontener(0, 0, "plastic")
     kontener_plastik.setImage(pygame.image.load(
-        "resources/plansza/pojemnik_plastik.png"))
+        "../resources/plansza/pojemnik_plastik.png"))
     plansza[0, 0].setJestKontenerem(True)
     plansza[0, 0].setObiekt(kontener_plastik)
 
     # domy
-    doms_array = ['resources/plansza/domy/dom1.png', 'resources/plansza/domy/dom2.png',
-                  'resources/plansza/domy/dom3.png', 'resources/plansza/domy/dom4.png',
-                  'resources/plansza/domy/dom5.png']
+    doms_array = ['../resources/plansza/domy/dom1.png', '../resources/plansza/domy/dom2.png',
+                  '../resources/plansza/domy/dom3.png', '../resources/plansza/domy/dom4.png',
+                  '../resources/plansza/domy/dom5.png']
 
     domy_lista = pygame.sprite.Group()
 

src/modele.py

@@ -25,7 +25,7 @@ class Smieciarka(pygame.sprite.Sprite):
     def __init__(self, x, y):
         self.x = x
         self.y = y
-        self.image = pygame.image.load('resources/plansza/smieciarka.png')
+        self.image = pygame.image.load('../resources/plansza/smieciarka.png')
         self.obraz = None
         self.ruch = 1
         self.plastik = []
@@ -79,7 +79,7 @@ class Smieciarka(pygame.sprite.Sprite):
                         self.zwiekszIloscOdwiedzonychDomow()
                 if self.ruch == 2:
                     self.image = pygame.image.load(
-                        'resources/plansza/smieciarka.png')
+                        '../resources/plansza/smieciarka.png')
                     self.ruch = 1
 
                 self.plansza[self.x - 1, self.y].setKolor(BLUE)

src/tworzenie_danych_AO.py

@@ -6,9 +6,9 @@ import os
 import h5py
 
 rozmiar_zdj = tuple((500, 500))
-sciezka_do_zdj = "resources/smieci_stare"
-h5_dane = 'parametry_zdjec.h5'
-h5_etykiety = 'etykiety.h5'
+sciezka_do_zdj = "../resources/dane/smieci_stare"
+h5_dane = '../resources/dane/parametry_zdjec.h5'
+h5_etykiety = '../resources/dane/etykiety.h5'
 
 
 def wyznaczHuMomenty(zdj):

src/uczenie_adamB.py

@@ -62,13 +62,13 @@ def train():
                 running_loss = 0.0
 
     print('Finished Training')
-    PATH = './wytrenowaned.pth'
+    PATH = '../resources/dane/wytrenowaned.pth'
     torch.save(net.state_dict(), PATH)
 
 
 def predict(img_path):
     net = Net()
-    PATH = './wytrenowaned.pth'
+    PATH = '../resources/dane/wytrenowaned.pth'
     img = Image.open(img_path)
     pil_to_tensor = transforms.ToTensor()(img).unsqueeze_(0)
     classes = ('glass', 'metal', 'paper', 'plastic')

src/uczenie_adamO.py

@@ -17,10 +17,10 @@ import pydotplus
 warnings.filterwarnings('ignore')
 
 rozmiar_zbioru_testowego = 0.20
-katalog_uczacy = "resources\\smieci_stare"
-katalog_testujacy = "resources\\smieci w kontenerach"
-h5_parametry = 'parametry_zdjec.h5'
-h5_etykiety = 'etykiety.h5'
+katalog_uczacy = "..\\resources\\smieci_stare"
+katalog_testujacy = "..\\resources\\smieci w kontenerach"
+h5_parametry = '..\\resources\\dane\\parametry_zdjec.h5'
+h5_etykiety = '..\\resources\\dane\\etykiety.h5'
 rozmiar_zdj = tuple((500, 500))
 
 

src/uczenie_kacper.py

@@ -9,8 +9,8 @@ from keras import backend as K
 
 img_width, img_height = 299, 299
 
-train_data_dir = 'resources/zbior_uczacy'
-validation_data_dir = 'resources/smieci'
+train_data_dir = '../resources/zbior_uczacy'
+validation_data_dir = '../resources/smieci'
 nb_train_samples = 1599
 nb_validation_samples = 395
 epochs = 1
@@ -79,12 +79,12 @@ def trainModel():
         validation_data=validation_generator,
         validation_steps=nb_validation_samples // batch_size,
         shuffle=True)
-    model.save_weights('nowy_wytrenowany.h5')
+    model.save_weights('../resources/dane/nowy_wytrenowany.h5')
 
 
 def przewidz(path):
     model = stworzModel()
-    model.load_weights('wytrenowany.h5')
+    model.load_weights('../resources/dane/wytrenowany.h5')
     img = load_img(path, target_size=(299, 299))
     img_array = img_to_array(img)
     img_array = np.expand_dims(img_array, axis=0)