refactor
This commit is contained in:
parent
10d90b5f65
commit
84e47b1c96
Before Width: | Height: | Size: 3.3 MiB After Width: | Height: | Size: 3.3 MiB |
@ -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)
|
@ -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()
|
||||||
|
|
@ -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)
|
@ -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):
|
@ -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')
|
@ -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))
|
||||||
|
|
||||||
|
|
@ -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)
|
Loading…
Reference in New Issue
Block a user