changes in common part
This commit is contained in:
parent
9e608db67c
commit
76ec19cde0
@ -20,6 +20,7 @@ from models.garbageDump import Dump
|
|||||||
|
|
||||||
|
|
||||||
def createSets():
|
def createSets():
|
||||||
|
if not path.exists('ClassificationGarbage/trainSet'):
|
||||||
rootDir = 'ClassificationGarbage'
|
rootDir = 'ClassificationGarbage'
|
||||||
typesDir = ['/cardboard', '/glass', '/metal', '/paper', '/plastic']
|
typesDir = ['/cardboard', '/glass', '/metal', '/paper', '/plastic']
|
||||||
testRatio = 0.2
|
testRatio = 0.2
|
||||||
|
Loading…
Reference in New Issue
Block a user