Merge pull request 'Merge with model for 2 crops only' (#23 ) from neural_network_two_crops into neural_network

Reviewed-on: #23
This: -added trained model for 2 crops -added way to choose which model should we use
2024-06-04 12:45:15 +02:00 · 2024-06-04 12:40:39 +02:00
2 changed files with 5 additions and 3 deletions
--- a/model_2_crops.pth
+++ b/model_2_crops.pth
--- a/neuralnetwork.py
+++ b/neuralnetwork.py
@ -9,12 +9,14 @@ 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("cuda" if torch.cuda.is_available() else "cpu")
 device = torch.device("cpu")
 # device = torch.device("mps") if torch.backends.mps.is_available() else torch.device('cpu')
 # print(device)
Author	SHA1	Message	Date
s481825	53ec8e993e	Merge pull request 'Merge with model for 2 crops only' (#23 ) from neural_network_two_crops into neural_network Reviewed-on: #23	2024-06-04 12:45:15 +02:00
jakzar	24724fb8b7	This: -added trained model for 2 crops -added way to choose which model should we use	2024-06-04 12:40:39 +02:00