From 84e47b1c96d84c869840d2317e1ee39c3fcc019d Mon Sep 17 00:00:00 2001 From: Adam Osiowy Date: Fri, 2 Apr 2021 22:39:43 +0200 Subject: [PATCH] refactor --- etykiety.h5 => resources/dane/etykiety.h5 | Bin .../dane/parametry_zdjec.h5 | Bin .../dane/wytrenowaned.pth | Bin .../dane/wytrenowany.h5 | Bin .../raporty/adamORaport.md | 0 .../raporty/environment.md | 0 graph.png => resources/raporty/graph.png | Bin .../raporty/kacperRaport.md | 0 .../raporty/raport_adamB.md | 0 .../raporty/report_koncowy.md | 0 .../raporty/route-planning.md | 0 astar.py => src/astar.py | 0 find.py => src/find.py | 4 +-- game.py => src/game.py | 26 +++++++++--------- main.py => src/main.py | 0 modele.py => src/modele.py | 4 +-- .../tworzenie_danych_AO.py | 6 ++-- uczenie_adamB.py => src/uczenie_adamB.py | 4 +-- uczenie_adamO.py => src/uczenie_adamO.py | 8 +++--- uczenie_kacper.py => src/uczenie_kacper.py | 8 +++--- 20 files changed, 30 insertions(+), 30 deletions(-) rename etykiety.h5 => resources/dane/etykiety.h5 (100%) rename parametry_zdjec.h5 => resources/dane/parametry_zdjec.h5 (100%) rename wytrenowaned.pth => resources/dane/wytrenowaned.pth (100%) rename wytrenowany.h5 => resources/dane/wytrenowany.h5 (100%) rename adamORaport.md => resources/raporty/adamORaport.md (100%) rename environment.md => resources/raporty/environment.md (100%) rename graph.png => resources/raporty/graph.png (100%) rename kacperRaport.md => resources/raporty/kacperRaport.md (100%) rename raport_adamB.md => resources/raporty/raport_adamB.md (100%) rename report_koncowy.md => resources/raporty/report_koncowy.md (100%) rename route-planning.md => resources/raporty/route-planning.md (100%) rename astar.py => src/astar.py (100%) rename find.py => src/find.py (76%) rename game.py => src/game.py (93%) rename main.py => src/main.py (100%) rename modele.py => src/modele.py (98%) rename tworzenie_danych_AO.py => src/tworzenie_danych_AO.py (92%) rename uczenie_adamB.py => src/uczenie_adamB.py (95%) rename uczenie_adamO.py => src/uczenie_adamO.py (95%) rename uczenie_kacper.py => src/uczenie_kacper.py (92%) diff --git a/etykiety.h5 b/resources/dane/etykiety.h5 similarity index 100% rename from etykiety.h5 rename to resources/dane/etykiety.h5 diff --git a/parametry_zdjec.h5 b/resources/dane/parametry_zdjec.h5 similarity index 100% rename from parametry_zdjec.h5 rename to resources/dane/parametry_zdjec.h5 diff --git a/wytrenowaned.pth b/resources/dane/wytrenowaned.pth similarity index 100% rename from wytrenowaned.pth rename to resources/dane/wytrenowaned.pth diff --git a/wytrenowany.h5 b/resources/dane/wytrenowany.h5 similarity index 100% rename from wytrenowany.h5 rename to resources/dane/wytrenowany.h5 diff --git a/adamORaport.md b/resources/raporty/adamORaport.md similarity index 100% rename from adamORaport.md rename to resources/raporty/adamORaport.md diff --git a/environment.md b/resources/raporty/environment.md similarity index 100% rename from environment.md rename to resources/raporty/environment.md diff --git a/graph.png b/resources/raporty/graph.png similarity index 100% rename from graph.png rename to resources/raporty/graph.png diff --git a/kacperRaport.md b/resources/raporty/kacperRaport.md similarity index 100% rename from kacperRaport.md rename to resources/raporty/kacperRaport.md diff --git a/raport_adamB.md b/resources/raporty/raport_adamB.md similarity index 100% rename from raport_adamB.md rename to resources/raporty/raport_adamB.md diff --git a/report_koncowy.md b/resources/raporty/report_koncowy.md similarity index 100% rename from report_koncowy.md rename to resources/raporty/report_koncowy.md diff --git a/route-planning.md b/resources/raporty/route-planning.md similarity index 100% rename from route-planning.md rename to resources/raporty/route-planning.md diff --git a/astar.py b/src/astar.py similarity index 100% rename from astar.py rename to src/astar.py diff --git a/find.py b/src/find.py similarity index 76% rename from find.py rename to src/find.py index cd28456..b00939d 100644 --- a/find.py +++ b/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) diff --git a/game.py b/src/game.py similarity index 93% rename from game.py rename to src/game.py index 803f977..dee92ff 100644 --- a/game.py +++ b/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() diff --git a/main.py b/src/main.py similarity index 100% rename from main.py rename to src/main.py diff --git a/modele.py b/src/modele.py similarity index 98% rename from modele.py rename to src/modele.py index 70d4269..94c78df 100644 --- a/modele.py +++ b/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) diff --git a/tworzenie_danych_AO.py b/src/tworzenie_danych_AO.py similarity index 92% rename from tworzenie_danych_AO.py rename to src/tworzenie_danych_AO.py index dae7527..df31d66 100644 --- a/tworzenie_danych_AO.py +++ b/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): diff --git a/uczenie_adamB.py b/src/uczenie_adamB.py similarity index 95% rename from uczenie_adamB.py rename to src/uczenie_adamB.py index 3f0e896..2c623ac 100644 --- a/uczenie_adamB.py +++ b/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') diff --git a/uczenie_adamO.py b/src/uczenie_adamO.py similarity index 95% rename from uczenie_adamO.py rename to src/uczenie_adamO.py index b20c690..a42980b 100644 --- a/uczenie_adamO.py +++ b/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)) diff --git a/uczenie_kacper.py b/src/uczenie_kacper.py similarity index 92% rename from uczenie_kacper.py rename to src/uczenie_kacper.py index 149c862..dc09648 100644 --- a/uczenie_kacper.py +++ b/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)