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

neural_network #25

Merged
s481834 merged 12 commits from neural_network into refactor 2024-06-04 13:23:20 +02:00
Conversation 0 Commits 12 Files Changed 14 +2 -2
3 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 5d7b68fdd6 - Show all commits

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)