Параметры модели предиктора формы dlib - PullRequest
0 голосов
/ 01 января 2019

Я натренировал модель предиктора формы dlib на своих пользовательских данных (используя файл train_shape_predictor.py).В результате я получил .dat файл.Теперь у меня есть изображение, содержащее объект, на котором обучена модель предсказания dlib.Как я буду использовать эту модель прогнозирования, чтобы предсказать форму на входном изображении?Я вижу документацию по прогнозированию формы Dlib , упоминается, что предиктор формы dlib принимает два аргумента:

  1. изображение
  2. прямоугольник (прямоугольник Dlib)

Теперь, какими будут эти параметры, в моем случае, поскольку у меня есть только одно изображение (содержащее объект, который будет предсказан с помощью обученной модели)?

Любая помощь в этомуважение будет высоко оценено.

1 Ответ

0 голосов
/ 01 января 2019

Как сказано в документе:

  • изображение - это крупный символ, содержащий 8-битное изображение в градациях серого или RGB-изображение.-> Передайте свое изображение сюда
  • - это ограничивающий прямоугольник, чтобы начать прогнозирование формы внутри.-> если у вас уже есть ограничивающая рамка вашего объекта (например, где находится лицо на изображении), передайте его здесь.

Типичное применение:

rects = dlib.simple_object_detector(image)

for rect in rects:
   shape = dlib.shape_predictor(image, rect)
...