neuralnetwork.py

@@ -9,14 +9,12 @@ from PIL import Image
 import random
 
 imageSize = (128, 128)
-labels = ['carrot','corn', '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', labels[3]:'obornik kurzy'}
-#labels = ['corn','tomato'] #uncomment this two lines for 2 crops only
-#fertilizer = {labels[0]: 'kompost', labels[1]: 'saletra amonowa'}
+
 torch.manual_seed(42)
 
-#device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-device = torch.device("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)