Ошибка OpenCV при сопоставлении функций с FLANN - PullRequest
Новая
       20

Ошибка OpenCV при сопоставлении функций с FLANN

0 голосов
/ 30 августа 2018

У меня есть фрагмент кода для сопоставления функций между фрагментом ролика и эталонным изображением. Обычно это работает хорошо, но иногда выдает ошибку в середине клипа. Поскольку он всегда находится в одних и тех же клипах, и в то же время, я думаю, что-то не так с кадром, который он пытается проанализировать.

Мой код: 

 cap = cv2.VideoCapture(clip_file)
 img1 = cv2.imread(ref_image,0)

 while(cap.isOpened()):

    # read the frame and convert to gray-scale
    ret, frame = cap.read()
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

    # Initiate ORB detector
    orb = cv2.ORB_create()

    # find the keypoints and descriptors with ORB
    kp1, des1 = orb.detectAndCompute(img1,None)
    kp2, des2 = orb.detectAndCompute(gray,None)

    # FLANN parameters
    FLANN_INDEX_LSH = 6
    index_params= dict(algorithm = FLANN_INDEX_LSH,
                table_number = 6, # 12
                key_size = 12,     # 20
                multi_probe_level = 1) #2
    search_params = dict(checks=50)   # or pass empty dictionary

    flann = cv2.FlannBasedMatcher(index_params,search_params)

    matches = flann.knnMatch(des1,des2,k=2)
    cv2.imshow('img3',frame)

Ошибка, которую иногда выдает во время воспроизведения клипа: 

Traceback (most recent call last):
 File "movie_test.py", line 81, in <module>
flann_movie('data/movie.avi','data/ref.jpg')
 File "movie_test.py", line 35, in flann_movie
matches = flann.knnMatch(des1,des2,k=1)
 cv2.error: OpenCV(3.4.2) C:\projects\opencv- 
  python\opencv\modules\flann\src\miniflann.cpp:317: error: (-5:Bad 
   argument) Only continuous arrays are supported in function 
   'cv::flann::buildIndex_'

Любые предложения о причинах ошибки будут оценены. Спасибо.

1 Ответ

0 голосов
/ 31 августа 2018

Я думаю, что ошибка вызвана кадрами вашего видео, в которых не было обнаружено никаких следов оригинального шаблона функции. Проверьте средние результаты сопоставления для каждого кадра, а затем, если это является причиной, измените параметры FLANN или просто пропустите эти кадры до появления ошибки.

...