Skip to content
GitLab
Menu
Projects
Groups
Snippets
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Frank Tjado Ihmels
medienverarbeitung17.projectmood
Commits
5545121a
Commit
5545121a
authored
Dec 06, 2017
by
tihmels
Browse files
Nachbesserung Sorted_set
parent
59735d39
Changes
1
Hide whitespace changes
Inline
Side-by-side
projectmood/sorted_set_facedetector.py
View file @
5545121a
...
...
@@ -31,16 +31,15 @@ def detect_faces(emotion):
fileNumber
=
0
for
f
in
files
:
frame
=
cv2
.
imread
(
f
)
# Open image
gray
=
cv2
.
cvtColor
(
frame
,
cv2
.
COLOR_BGR2GRAY
)
# Convert image to grayscale
# Detect face using 4 different classifiers
face
=
faceDet
.
detectMultiScale
(
g
ra
y
,
scaleFactor
=
1.05
,
minNeighbors
=
6
,
minSize
=
(
10
,
10
),
face
=
faceDet
.
detectMultiScale
(
f
ra
me
,
scaleFactor
=
1.05
,
minNeighbors
=
6
,
minSize
=
(
10
,
10
),
flags
=
cv2
.
CASCADE_SCALE_IMAGE
)
face_two
=
faceDet_two
.
detectMultiScale
(
g
ra
y
,
scaleFactor
=
1.05
,
minNeighbors
=
6
,
minSize
=
(
10
,
10
),
face_two
=
faceDet_two
.
detectMultiScale
(
f
ra
me
,
scaleFactor
=
1.05
,
minNeighbors
=
6
,
minSize
=
(
10
,
10
),
flags
=
cv2
.
CASCADE_SCALE_IMAGE
)
face_three
=
faceDet_three
.
detectMultiScale
(
g
ra
y
,
scaleFactor
=
1.05
,
minNeighbors
=
6
,
minSize
=
(
10
,
10
),
face_three
=
faceDet_three
.
detectMultiScale
(
f
ra
me
,
scaleFactor
=
1.05
,
minNeighbors
=
6
,
minSize
=
(
10
,
10
),
flags
=
cv2
.
CASCADE_SCALE_IMAGE
)
face_four
=
faceDet_four
.
detectMultiScale
(
g
ra
y
,
scaleFactor
=
1.05
,
minNeighbors
=
6
,
minSize
=
(
10
,
10
),
face_four
=
faceDet_four
.
detectMultiScale
(
f
ra
me
,
scaleFactor
=
1.05
,
minNeighbors
=
6
,
minSize
=
(
10
,
10
),
flags
=
cv2
.
CASCADE_SCALE_IMAGE
)
# Go over detected faces, stop at first detected face, return empty if no face.
...
...
@@ -54,15 +53,16 @@ def detect_faces(emotion):
facefeatures
=
face_four
else
:
undetected
.
append
(
f
)
facefeatures
=
''
# Cut and save face
for
(
x
,
y
,
w
,
h
)
in
facefeatures
:
# get coordinates and size of rectangle containing face
logging
.
debug
(
"face found in file: {}"
.
format
(
f
))
totalFaces
+=
1
g
ra
y
=
g
ra
y
[
y
:
y
+
h
,
x
:
x
+
w
]
# Cut the frame to size
f
ra
me
=
f
ra
me
[
y
:
y
+
h
,
x
:
x
+
w
]
# Cut the frame to size
try
:
out
=
cv2
.
resize
(
g
ra
y
,
(
250
,
250
))
# Resize face so all images have same size
out
=
cv2
.
resize
(
f
ra
me
,
(
250
,
250
))
# Resize face so all images have same size
cv2
.
imwrite
(
'Basis_data/dataset/%s/%s.jpg'
%
(
emotion
,
fileNumber
),
out
)
# Write image
except
:
pass
# If error, pass file
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment