Compare commits
No commits in common. "53ec8e993edfe6000fadda0339a07c39ec082d4f" and "38c1adb0541bf7f2cacb358a6b27f89cb9b937dd" have entirely different histories.
53ec8e993e
...
38c1adb054
Binary file not shown.
@ -9,14 +9,12 @@ from PIL import Image
|
|||||||
import random
|
import random
|
||||||
|
|
||||||
imageSize = (128, 128)
|
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'}
|
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)
|
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')
|
# device = torch.device("mps") if torch.backends.mps.is_available() else torch.device('cpu')
|
||||||
# print(device)
|
# print(device)
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user