Leggi frame da RTSP Stream in Python

Question

Ho recentemente installato una fotocamera Raspberry Pi e sto trasmettendo i frame su RTSP. Anche se potrebbe non essere completamente necessario, ecco il comando che sto utilizzando per trasmettere il video:

raspivid -o - -t 0 -w 1280 -h 800 |cvlc -vvv stream:///dev/stdin --sout '#rtp{sdp=rtsp://:8554/output.h264}' :demux=h264 

Questo streaming del video perfettamente.

Quello che vorrei fare ora è analizzare questo flusso con Python e leggere ogni frame singolarmente. Mi piacerebbe fare alcuni rilevamenti di movimento a scopo di sorveglianza.

Sono completamente perso su dove iniziare in questa attività. Qualcuno può indicarmi un buon tutorial? Se questo non è possibile tramite Python, quali strumenti/lingue posso usare per realizzare questo?

Answer 1

A seconda del tipo di flusso, è possibile dare un'occhiata a questo progetto per alcune idee.

https://code.google.com/p/python-mjpeg-over-rtsp-client/

Se vuoi essere mega-pro, si potrebbe usare qualcosa di simile (moduli Python disponibili credo) http://opencv.org/ per la gestione del rilevamento del movimento.

Answer 2

Bit di una soluzione hacky, ma è possibile utilizzare la VLC python bindings e riprodurre il flusso:

player=vlc.MediaPlayer('rtsp://:8554/output.h264') 
player.play() 

Poi scattare una foto ogni secondo o giù di lì:

while 1: 
    time.sleep(1) 
    player.video_take_snapshot(0, '.snapshot.tmp.png', 0, 0) 

E quindi è possibile utilizzare SimpleCV o qualcosa da elaborare (basta caricare il file di immagine '.snapshot.tmp.png' nella libreria di elaborazione).

Answer 3

È possibile ottenere fotogrammi di lettura dal video utilizzando Python e OpenCV. Di seguito è riportato il codice di esempio. Funziona bene con la versione python e opencv2.

import cv2 
import os 
#Below code will capture the video frames and will sve it a folder (in current working directory) 

dirname = 'myfolder' 
#video path 
cap = cv2.VideoCapture("TestVideo.mp4") 
count = 0 
while(cap.isOpened()): 
    ret, frame = cap.read() 
    if not ret: 
     break 
    else: 
     cv2.imshow('frame', frame) 
     #The received "frame" will be saved. Or you can manipulate "frame" as per your needs. 
     name = "rec_frame"+str(count)+".jpg" 
     cv2.imwrite(os.path.join(dirname,name), frame) 
     count += 1 
    if cv2.waitKey(20) & 0xFF == ord('q'): 
     break 
cap.release() 
cv2.destroyAllWindows()