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

This: -removed model.pth from git tracking for now -added corn to dataset

Browse Source
This commit is contained in:
jakzar 2024-05-26 14:46:12 +02:00
parent 34bfe78218
commit 5d7b68fdd6
3 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
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=False newModel=True
pygame.init() pygame.init()
show_console=True show_console=True

BIN
model.pth
View File

Binary file not shown.

2
neuralnetwork.py
View File

@ -9,7 +9,7 @@ from PIL import Image
import random import random
imageSize = (128, 128) imageSize = (128, 128)
labels = ['carrot', 'potato', 'tomato'] # musi być w kolejności alfabetycznej labels = ['carrot','corn', 'potato', 'tomato'] # musi być w kolejności alfabetycznej
fertilizer = {labels[0]: 'kompost', labels[1]: 'saletra amonowa', labels[2]: 'superfosfat'} fertilizer = {labels[0]: 'kompost', labels[1]: 'saletra amonowa', labels[2]: 'superfosfat'}
torch.manual_seed(42) torch.manual_seed(42)