OpenCV Python une iteraciones increíblemente lentas a través de datos de imagen

Question

Recientemente tomé un código que rastreó un objeto basado en el color en OpenCV C++ y lo reescribí en los enlaces de python.

Los resultados generales y el método fueron la misma sintaxis negativa, obviamente. Pero, cuando realizo el siguiente código en cada fotograma de un video, me toma casi 2-3 segundos completarlo, ya que la variante de C++, también a continuación, es instantánea en comparación y puedo iterar entre fotogramas tan rápido como mi dedo puede presionar un llave.

¿Alguna idea o comentario?

cv.PyrDown(img, dsimg) 
    for i in range(0, dsimg.height): 
     for j in range(0, dsimg.width): 

      if dsimg[i,j][1] > (_RED_DIFF + dsimg[i,j][2]) and dsimg[i,j][1] > (_BLU_DIFF + dsimg[i,j][0]): 
       res[i,j] = 255 
      else: 
       res[i,j] = 0 

---

for(int i =0; i < (height); i++) 
    { 
     for(int j = 0; j < (width); j++) 
     { 
      if(((data[i * step + j * channels + 1]) > (RED_DIFF + data[i * step + j * channels + 2])) && 
       ((data[i * step + j * channels + 1]) > (BLU_DIFF + data[i * step + j * channels]))) 
       data_r[i *step_r + j * channels_r] = 255; 
      else 
       data_r[i * step_r + j * channels_r] = 0; 
     } 
    } 

Gracias

Answer 1

Trate de usar numpy para hacer su cálculo, en lugar de bucles anidados. Debería obtener un rendimiento tipo C para cálculos simples como este de numpy.

Por ejemplo, su anidado para los bucles pueden ser reemplazados con un par de expresiones numpy ...

No estoy terriblemente familiar con OpenCV, pero creo que los enlaces Python ahora tienen una interfaz serie numpy, por lo el ejemplo anterior debería ser tan simple como:

cv.PyrDown(img, dsimg) 

data = np.asarray(dsimg) 
blue, green, red = data.T 

res = (green > (_RED_DIFF + red)) & (green > (_BLU_DIFF + blue)) 
res = res.astype(np.uint8) * 255 

res = cv.fromarray(res) 

(no está comprobado, por supuesto ...) Una vez más, estoy realmente no muy familiarizados con OpenCV, pero pitón bucles for anidados no son la manera de ir sobre la modificando una imagen por elementos, independientemente ...

Espero que ayude un poco, de todos modos!