Рисование результатов calcOpticalFlowFarneback в OpenCV без цикла

Question

Я использую cv2.calcOpticalFlowFarneback() в python для получения результатов оптического потока, но есть проблема при попытке использовать его результаты. Я хочу нарисовать изменение, возвращаемое этой функцией, но цикл в python слишком медленный. Я уверен, что есть метод с numpy, чтобы сделать его быстрее. Вот моя попытка

flow = cv2.calcOpticalFlowFarneback(prev_gray, gray, pyr_scale=0.5, levels=5, winsize=13, iterations=10, poly_n=5, poly_sigma=1.1, flags=0,flow=None) 

for y in range(0, flow.shape[0], 5): 

    for x in range(0, flow.shape[1], 5): 

     fx, fy = flow[y, x] 

     cv2.line(frame2, (x,y), (int(x+fx),int(y+fy)), (255,0,0)) 
     cv2.circle(frame2, (x,y), 1, (255,0,0), -1) 

Answer 1

Вы можете использовать метод OpenCV cv2.polylines(), чтобы нарисовать массив ломаных без итерировать с цикл.

здесь пример кода

def draw_flow(self, img, flow, step=16): 
    h, w = img.shape[:2] 
    y, x = np.mgrid[step/2:h:step, step/2:w:step].reshape(2, -1).astype(int) 
    fx, fy = flow[y, x].T 
    lines = np.vstack([x, y, x + fx, y + fy]).T.reshape(-1, 2, 2) 
    lines = np.int32(lines + 0.5) 
    vis = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR) 
    cv2.polylines(vis, lines, 0, (0, 255, 0)) 
    for (x1, y1), (x2, y2) in lines: 
     cv2.circle(vis, (x1, y1), 1, (0, 255, 0), -1) 
    return vis 

проверка из образцов OpenCV, чтобы загрузить и запустить полный код https://github.com/opencv/opencv/blob/master/samples/python/opt_flow.py

или документации OpenCV, чтобы получить больше информации о полилинии http://docs.opencv.org/3.0-beta/modules/imgproc/doc/drawing_functions.html#polylines