[python] ネットワークカメラの画像を RTSP で取得して表示・保存

2021/08/03

[python] ネットワークカメラの画像を RTSP で取得して表示・保存

## 画像を取得して表示

`VideoCapture` の引数に RTSP の URL を記述するだけで簡単に画像を取得できます。  
URL はネットワークカメラの仕様を確認してください。  
今回は Panasonic のカメラを使いました。

```py
import cv2

user_id = "admin" 
user_pw = "password" 
host = "192.168.0.1" 
cap = cv2.VideoCapture(f"rtsp://{user_id}:{user_pw}@{host}/MediaInput/h264")

while(True):
    try:
        ret, frame = cap.read()
        if ret == True:
            cv2.imshow('VIDEO', frame)
        cv2.waitKey(1)
    except KeyboardInterrupt:
        print("KeyboardInterrupt")
        break

cap.release()
cv2.destroyAllWindows()
```

VideoCapture の引数を変更すれば、USB カメラや動画ファイルの読み込み、連番のJPGファイルも一括で読み込めたりするので便利です。

## 保存

今回 RTSP では H.264 で取得しましたが、フレームの処理は OpenCV 側で上手く処理してくれているので、キーフレームとか GOP とか何も考えずに保存できます。

### JPG の保存

```
cv2.imwrite("sample.jpg", frame)
```

### 動画の保存

`VideoWriter_fourcc` でコーデックを指定します。  
以下が参考になりました。

- https://qiita.com/Gyutan/items/e279e8680b90ca891250

画像サイズはキャプチャから取得できる値をそのまま保存時にも使っていますが、フレームレートは取得できないので、直接指定しています。  
これが合っていないと当然おかしなことになります（早送り再生やスロー再生したような状態になります）。
  
```
import cv2

user_id = "admin" 
user_pw = "password" 
host = "192.168.0.1" 
cap = cv2.VideoCapture(f"rtsp://{user_id}:{user_pw}@{host}/MediaInput/h264")

width = cap.get(cv2.CAP_PROP_FRAME_WIDTH)
height = cap.get(cv2.CAP_PROP_FRAME_HEIGHT)

fmt = cv2.VideoWriter_fourcc(*'MP4V')
frame_rate = 30.0
writer = cv2.VideoWriter("sample.mp4", fmt, frame_rate, (int(width), int(height)))

while(True):
    try:
        ret, frame = cap.read()
        if ret == True:
            writer.write(frame)
            cv2.imshow('VIDEO', frame)
        cv2.waitKey(1)
    except KeyboardInterrupt:
        print("KeyboardInterrupt")
        break

writer.release()
cap.release()
cv2.destroyAllWindows()
```

へっぽこプログラマーの備忘録

2021/08/03

[python] ネットワークカメラの画像を RTSP で取得して表示・保存

Translate

Popular Posts

Labels

Blog Archive