s473725/Madra_smieciarka
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

NN + DT connection

Browse Source 
Signed-off-by: Neerka <kuba.markil0220@gmail.com>
This commit is contained in:
Neerka 2024-05-27 10:36:48 +02:00
parent 18cac8bca9
commit 4a6045bdc9
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
classes/Trash.py
View File

@ -1,11 +1,14 @@
from random import choice as rd from random import choice as rd
import numpy as np
from classes.trashFunctions import generatePicList from classes.trashFunctions import generatePicList
class Trash: class Trash:
tdic = {"paper": 1,
"bio_waste": 0,
"mixed": 4,
"glass": 3,
"metals_and_plastics": 2}
def __init__(self): def __init__(self):
self.tree_type = 0 self.tree_type = 0
self.ttype = "Trash" self.ttype = "Trash"
@ -24,7 +27,7 @@ class Trash:
return self.tree_type return self.tree_type
def setTreetype(self, treetype): def setTreetype(self, treetype):
self.tree_type = id self.tree_type = treetype
return self return self
def getTtype(self): def getTtype(self):
@ -32,6 +35,7 @@ class Trash:
def setTtype(self, ttype): def setTtype(self, ttype):
self.ttype = ttype self.ttype = ttype
self.setTreetype(self.tdic[ttype])
return self return self
def getWaga(self): def getWaga(self):