s444517/AI2020_Project
1
2
Fork 0
Code Issues Pull Requests Releases Wiki Activity

changes in common part

Browse Source
This commit is contained in:
Kamila Bobkowska 2020-06-04 07:33:58 +00:00
parent 9e608db67c
commit 76ec19cde0
1 changed files with 232 additions and 231 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
garbageDumpSorting.py
View File

@ -20,29 +20,30 @@ from models.garbageDump import Dump
def createSets():
rootDir = 'ClassificationGarbage'
typesDir = ['/cardboard', '/glass', '/metal', '/paper', '/plastic']
testRatio = 0.2
if not path.exists('ClassificationGarbage/trainSet'):
rootDir = 'ClassificationGarbage'
typesDir = ['/cardboard', '/glass', '/metal', '/paper', '/plastic']
testRatio = 0.2
for cls in typesDir:
os.makedirs(rootDir + '/trainSet' + cls)
os.makedirs(rootDir + '/testSet' + cls)
sourceDir = rootDir + cls
allFileNames = os.listdir(sourceDir)
np.random.shuffle(allFileNames)
trainingFileNames, testFileNames = np.split(np.array(allFileNames), [int(len(allFileNames) * (1 - testRatio))])
trainingFileNames = [sourceDir +'/' + name for name in trainingFileNames.tolist()]
testFileNames = [sourceDir +'/' + name for name in testFileNames.tolist()]
print(cls + ':')
print('Total images: ', len(allFileNames))
print('Training: ', len(trainingFileNames))
print('Testing: ', len(testFileNames))
for cls in typesDir:
os.makedirs(rootDir + '/trainSet' + cls)
os.makedirs(rootDir + '/testSet' + cls)
sourceDir = rootDir + cls
allFileNames = os.listdir(sourceDir)
np.random.shuffle(allFileNames)
trainingFileNames, testFileNames = np.split(np.array(allFileNames), [int(len(allFileNames) * (1 - testRatio))])
trainingFileNames = [sourceDir +'/' + name for name in trainingFileNames.tolist()]
testFileNames = [sourceDir +'/' + name for name in testFileNames.tolist()]
print(cls + ':')
print('Total images: ', len(allFileNames))
print('Training: ', len(trainingFileNames))
print('Testing: ', len(testFileNames))
for name in trainingFileNames:
shutil.copy(name, rootDir +'/trainSet' + cls)
for name in testFileNames:
shutil.copy(name, rootDir +'/testSet' + cls)
print("Images copied.")
for name in trainingFileNames:
shutil.copy(name, rootDir +'/trainSet' + cls)
for name in testFileNames:
shutil.copy(name, rootDir +'/testSet' + cls)
print("Images copied.")
def processTrainData():