diff --git a/samplelib/Sample.py b/samplelib/Sample.py index 58c92d5..80910d5 100644 --- a/samplelib/Sample.py +++ b/samplelib/Sample.py @@ -61,7 +61,7 @@ class Sample(object): def get_pitch_yaw_roll(self): if self.pitch_yaw_roll is None: - self.pitch_yaw_roll = LandmarksProcessor.estimate_pitch_yaw_roll(landmarks) + self.pitch_yaw_roll = LandmarksProcessor.estimate_pitch_yaw_roll(landmarks, size=self.shape[1]) return self.pitch_yaw_roll def set_filename_offset_size(self, filename, offset, size):