s434720/2019SZI-Projekt
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add new operations

Browse Source
This commit is contained in:
nlitkowski 2019-05-05 16:56:09 +02:00
parent 68e49591f7
commit 158c26ec01
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
Logic/TrashRecognition/AugmentImages.py
View File

@ -1,18 +1,20 @@
def main():
import Augmentor
p = [Augmentor.Pipeline("Images/TrainingImages/glass"),
Augmentor.Pipeline("Images/TrainingImages/metal"),
Augmentor.Pipeline("Images/TrainingImages/paper"),
Augmentor.Pipeline("Images/TrainingImages/plastic")]
for i in range(len(p)):
# Setting rotate values
p[i].rotate(probability=0.7, max_left_rotation=10, max_right_rotation=10)
# Setting zoom values
for i in range(len(p)):
# Setting the operations to perform on image set
p[i].rotate(probability=0.7, max_left_rotation=10, max_right_rotation=10)
p[i].zoom(probability=0.5, min_factor=1.1, max_factor=1.5)
p[i].flip_random(probability=0.6)
p[i].skew(probability=0.7, magnitude=0.7)
# Size of training sample
p[i].sample(1000)
p[i].sample(2000)
p[i].process()