NN + DT connection
Signed-off-by: Neerka <kuba.markil0220@gmail.com>
This commit is contained in:
parent
18cac8bca9
commit
4a6045bdc9
@ -1,11 +1,14 @@
|
||||
from random import choice as rd
|
||||
|
||||
import numpy as np
|
||||
|
||||
from classes.trashFunctions import generatePicList
|
||||
|
||||
|
||||
class Trash:
|
||||
tdic = {"paper": 1,
|
||||
"bio_waste": 0,
|
||||
"mixed": 4,
|
||||
"glass": 3,
|
||||
"metals_and_plastics": 2}
|
||||
|
||||
def __init__(self):
|
||||
self.tree_type = 0
|
||||
self.ttype = "Trash"
|
||||
@ -24,7 +27,7 @@ class Trash:
|
||||
return self.tree_type
|
||||
|
||||
def setTreetype(self, treetype):
|
||||
self.tree_type = id
|
||||
self.tree_type = treetype
|
||||
return self
|
||||
|
||||
def getTtype(self):
|
||||
@ -32,6 +35,7 @@ class Trash:
|
||||
|
||||
def setTtype(self, ttype):
|
||||
self.ttype = ttype
|
||||
self.setTreetype(self.tdic[ttype])
|
||||
return self
|
||||
|
||||
def getWaga(self):
|
||||
|
Loading…
Reference in New Issue
Block a user