изменить размеры видео входа для face_detection - PullRequest
Новая
       27

изменить размеры видео входа для face_detection

0 голосов
/ 09 марта 2020

Я развернул функцию обнаружения лица на видео, очень внимательно следя за репо здесь (https://gist.github.com/fyr91/79aaf4b6d679814406ee4028bd03b7aa)

Модель проходит, но когда я вижу видеопоток, она предсказывает, что это займет только часть моего видео (вверху слева), и я локализую его для этой части кода 

# preprocess img acquired
        img = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) # convert bgr to rgb
        img = cv2.resize(img, (640, 480)) # resize

Я понял, что входное видео, которое я пытаюсь вписать, имеет размеры 3840 x 2160, поэтому при удалении изменения размера выдается следующая ошибка: 

InvalidArgument: [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Got invalid dimensions for input: input for the following indices
 index: 2 Got: 2160 Expected: 480
 index: 3 Got: 3840 Expected: 640
 Please fix either the inputs or the model.

Мне интересно, есть ли способ исправить входы для достижения желаемого разрешения. Спасибо!

...