This commit is contained in:
Adam Osiowy 2021-04-02 22:39:43 +02:00
parent 10d90b5f65
commit 84e47b1c96
20 changed files with 30 additions and 30 deletions

View File

Before

Width:  |  Height:  |  Size: 3.3 MiB

After

Width:  |  Height:  |  Size: 3.3 MiB

View File

@ -11,9 +11,9 @@ from PIL import Image
import os 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) pil_to_tensor = transforms.ToTensor()(img).unsqueeze_(0)
if(str(pil_to_tensor.shape[1]) == '1'): if(str(pil_to_tensor.shape[1]) == '1'):
print(filename) print(filename)

View File

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

View File

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

View File

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

View File

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

View File

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

View File

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