This: -removed model.pth from git tracking for now -added corn to dataset
This commit is contained in:
parent
34bfe78218
commit
5d7b68fdd6
2
App.py
2
App.py
@ -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
|
||||
|
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user