Commit 7025d572 authored by Olegs Nikisins's avatar Olegs Nikisins
Browse files

Fixed the typo

parent ae32f073
Pipeline #17303 passed with stage
in 22 minutes and 59 seconds
......@@ -85,7 +85,7 @@ class Wrapper(bob.bio.base.preprocessor.Preprocessor):
self.quality_function = quality_function
self.compressed_io = compressed_io
sels.quality_assessment_function = quality_assessment_function
self.quality_assessment_function = quality_assessment_function
def _check_data(self, frames):
"""Checks if the given video is in the desired format."""
......@@ -121,6 +121,8 @@ class Wrapper(bob.bio.base.preprocessor.Preprocessor):
annots = None
fc = utils.FrameContainer()
quality_index = 0
for index, frame, _ in frames:
# if annotations are given, we take them
if annotations is not None: annots = annotations[index]
......@@ -140,7 +142,8 @@ class Wrapper(bob.bio.base.preprocessor.Preprocessor):
if self.quality_assessment_function(quality): #quality satisfies our criteria
fc.add(index, preprocessed, quality)
fc.add(quality_index, preprocessed) # no need to add quality here because we already addressed it
quality_index += 1
else:
fc.add(index, preprocessed, quality)
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment