Merge pull request 'add garbage class' (#8) from garbage into master
Reviewed-on: #8 Reviewed-by: Mikołaj Gawor <mikgaw@st.amu.edu.pl>
This commit is contained in:
commit
94e65feaa9
21
garbage.py
Normal file
21
garbage.py
Normal file
@ -0,0 +1,21 @@
|
||||
from enum import Enum
|
||||
|
||||
class GarbageType (Enum):
|
||||
PAPER = 0,
|
||||
PLASTIC_AND_METAL = 1
|
||||
GLASS = 3
|
||||
BIO = 4
|
||||
MIXED = 5
|
||||
|
||||
class Garbage:
|
||||
img: str
|
||||
|
||||
def __init__(self, img: str) -> None:
|
||||
self.img = img
|
||||
|
||||
class RecognizedGarbage (Garbage):
|
||||
garbage_type: GarbageType
|
||||
|
||||
def __init__(self, src: Garbage, garbage_type: GarbageType) -> None:
|
||||
super().__init__(src.img)
|
||||
self.garbage_type = garbage_type
|
Loading…
Reference in New Issue
Block a user