OpenCV с NVIDIA VPI (интерфейс программирования зрения), ядро ​​сброшено ⇐ Python

[Anonymous]

Я играю с интерфейсом программирования Vision (VPI) и пытаюсь согнуть изображения. Я наткнулся на этот пример коррекции искажений объектива (https://docs.nvidia.com/vpi/algo_ldc.html) и добавил некоторый код, чтобы оно приняло входное изображение и показывает неискаженное выходное изображение. Следующий код работает нормально, и я могу просматривать выходное изображение. Как только я не пошутил, строка «VideoCapture» я получаю следующую ошибку: < /p>
»ошибка сегментации (ядро сброшено). < /P>
Кто -нибудь может помочь мне использовать этот код для видео ввода? < /P>
import vpi
import numpy as np
import cv2
import PIL
from PIL import Image

img = cv2.imread('input.jpeg')

#cap = cv2.VideoCapture(0)

vpi_image = vpi.asimage(np.asarray(img))

grid = vpi.WarpGrid((2064,1544))

sensorWidth = 7.12
focallength = 3.5

f = focallength * (2064 / sensorWidth)

K = [[f, 0, 2064/2],
[0, f, 1544/2]]

X = np.eye(3,4)

warp = vpi.WarpMap.fisheye_correction(grid, K=K, X=X,
mapping=vpi.FisheyeMapping.EQUIDISTANT,
coeffs=[-0.01, 0.22])

with vpi.Backend.CUDA:
output = vpi_image.remap(warp, interp=vpi.Interp.CATMULL_ROM, border=vpi.Border.ZERO)

with output.rlock():
output = Image.fromarray(output.cpu()).save('output.jpeg')

pil_image = PIL.Image.open('output.jpeg').convert('RGB')
cv2_image = np.array(pil_image)
cv2_image = cv2_image[:, :, ::-1].copy()

cv2_image = cv2.resize(cv2_image, (920,590))
img = cv2.resize(img, (920, 590))

sbs = cv2.hconcat([img, cv2_image])
cv2.imshow("sbs", sbs)
cv2.waitKey(0)


Подробнее здесь: https://stackoverflow.com/questions/733 ... ore-dumped