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

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

2
App.py
View File

@ -21,7 +21,7 @@ if bfs3_flag or Astar or Astar2:
Pole.stoneFlag = True
TreeFlag=False
nnFlag=True
newModel=False
newModel=True
pygame.init()
show_console=True

BIN
model.pth

Binary file not shown.

View File

@ -9,7 +9,7 @@ from PIL import Image
import random
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'}
torch.manual_seed(42)