Определение позы с использованием предварительно обученных моделей в наборе данных COCO - PullRequest
0 голосов
/ 04 мая 2020

В настоящее время я использую относящиеся к Open Pose модели из набора данных Coco, чтобы предсказать ключевые точки для изображения, которое у меня есть. Я использую коды из следующих двух ссылок, чтобы опробовать определение позы на имеющихся у меня пользовательских данных.

https://www.learnopencv.com/deep-learning-based-human-pose-estimation-using-opencv-cpp-python/ https://github.com/spmallick/learnopencv/blob/master/OpenPose/OpenPose_Notebook.ipynb

На входе у меня есть человек, который тренируется лежа. Я дал изображение ниже: Input image

Однако обнаруженные ключевые точки не включают несколько точек в теле, таких как колено, как показано на рисунке ниже:

Output image

Может кто-нибудь дать подсказки, как это улучшить? или есть лучший набор данных или другая предварительно обученная модель, которая может обеспечить лучшую точность? Спасибо.

...