s481825/AI_PROJECT
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Trained the model.

Browse Source
This commit is contained in:
Mateusz Czajka 2024-05-26 18:41:55 +02:00
parent 5d7b68fdd6
commit ccd1cb2359
4 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
App.py
View File

@ -21,7 +21,7 @@ if bfs3_flag or Astar or Astar2:
Pole.stoneFlag = True Pole.stoneFlag = True
TreeFlag=False TreeFlag=False
nnFlag=True nnFlag=True
newModel=True newModel=False
pygame.init() pygame.init()
show_console=True show_console=True
@ -124,7 +124,7 @@ def init_demo(): #Demo purpose
if (newModel): if (newModel):
print_to_console("uczenie sieci neuronowej") print_to_console("uczenie sieci neuronowej")
model = neuralnetwork.trainNewModel() model = neuralnetwork.trainNewModel()
neuralnetwork.saveModel(model, 'model2.pth') neuralnetwork.saveModel(model, 'model.pth')
print_to_console("sieć nuronowa nauczona") print_to_console("sieć nuronowa nauczona")
print('model został wygenerowany') print('model został wygenerowany')
else: else:

7
Tractor.py
View File

@ -202,7 +202,12 @@ class Tractor:
quit() quit()
if self.slot.imagePath != None: if self.slot.imagePath != None:
predictedLabel = nn.predictLabel(self.slot.imagePath, model) predictedLabel = nn.predictLabel(self.slot.imagePath, model)
print(str("Coords: ({:02d}, {:02d})").format(self.slot.x_axis, self.slot.y_axis), "real:", self.slot.label, "predicted:", predictedLabel, "correct" if (self.slot.label == predictedLabel) else "incorrect", 'nawożę za pomocą:', nn.fertilizer[predictedLabel])
# print(str("Coords: ({:02d}, {:02d})").format(self.slot.x_axis, self.slot.y_axis), "real:", self.slot.label, "predicted:", predictedLabel, "correct" if (self.slot.label == predictedLabel) else "incorrect", 'nawożę za pomocą:', nn.fertilizer[predictedLabel])
if str(self.slot.label) != str(predictedLabel):
print(str("Coords: ({:02d}, {:02d})").format(self.slot.x_axis, self.slot.y_axis))
print("real: ", str(self.slot.label), "\tpredicted: ", str(predictedLabel), "\n")
if self.slot.label != predictedLabel: if self.slot.label != predictedLabel:
self.slot.mark_visited() self.slot.mark_visited()
count += 1 count += 1

BIN
model.pth Normal file
View File

Binary file not shown.

2
neuralnetwork.py
View File

@ -15,6 +15,8 @@ fertilizer = {labels[0]: 'kompost', labels[1]: 'saletra amonowa', labels[2]: 'su
torch.manual_seed(42) torch.manual_seed(42)
device = torch.device("cuda" if torch.cuda.is_available() else "cpu") device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
# device = torch.device("mps") if torch.backends.mps.is_available() else torch.device('cpu')
# print(device)
def getTransformation(): def getTransformation():
transform=transforms.Compose([ transform=transforms.Compose([