MLKit поворачивает изображение лица так, чтобы оно было ровным (iOS и Android)

Question

Я разрабатываю приложение, в котором я использую Firebase MLKit для обнаружения лиц. Затем я беру обнаруженное лицо и пропускаю его через пользовательскую модель тензорного потока.

Проблема в том, что моя пользовательская модель тензорного потока ожидает, что грани будут ровными. Если лицо пользователя вообще повернуто, я не получаю точных результатов

Есть ли способ повернуть изображение лица так, чтобы оно было прямым? Как мне это сделать с помощью MLKit?

Мой текущий способ получить изображение лица - использовать AVCaptureSession (iOS) и вырезать изображение на основе границ, данных мне в объекте лица из mlkit. Это работает, но опять же, если лицо пользователя не прямое, у меня получается повернутое лицо.

Answer 1

Вы можете использовать эту степень вращения головы, чтобы повернуть голову назад. Но это может работать только с EulerAngleZ(), когда голова вращается в плоскости XY.