TP-Link|ネットワークカメラのRTSP映像を受信するexeファイルの作成

コードレビュー

1. 全体的な印象

• 要件を満たすコードであることは確認できました。
• iniファイルからRTSP URLを読み込み、4つのストリームを別々のVideoCaptureオブジェクトで取得している点、フレームサイズをウィンドウサイズに合わせてリサイズしている点など、良い実装です。
• いくつかの改善点と、潜在的な問題点が見受けられますので、以下に詳細を説明します。

2. 具体的な改善点

2.1 エラー処理

• VideoCapture.isOpened()を使用して、ストリームが開けなかった場合のエラー処理を行っていますが、エラーメッセージにストリーム番号が含まれていません。
  • どのストリームが開けなかったのかを特定できるように、エラーメッセージにストリーム番号を含めるようにしましょう。
  • 例：print(f"Failed to open stream {i+1}: {streams[i]}")

2.2 フレーム取得のエラー処理

• cap.read()でフレームを取得していますが、retがFalseの場合のエラー処理がありません。
  • フレーム取得に失敗した場合も、適切なエラー処理を行うようにしましょう。
  • 例：if not ret: continue

2.3 ウィンドウサイズの取得

• cv2.getWindowImageRect('RTSP Streams')を使用してウィンドウサイズを取得していますが、この関数は非推奨です。
  • 代わりに、cv2.getWinProperty('RTSP Streams', cv2.WND_PROP_FRAME_WIDTH)とcv2.getWinProperty('RTSP Streams', cv2.WND_PROP_FRAME_HEIGHT)を使用して、フレームの幅と高さを取得するようにしましょう。

2.4 フレームのリサイズ

• フレームのリサイズ処理で、np.hstackとnp.vstackを使用していますが、この方法ではメモリ使用量が多くなります。
  • cv2.resizeとcv2.hconcatを使用して、直接結合する方法の方が効率的です。

3. 潜在的な問題点

3.1 マルチスレッド処理

• 4つのストリームを同時に処理していますが、マルチスレッド処理ではないため、CPU使用率が高くなる可能性があります。
  • 処理速度が遅くなる場合、マルチスレッド処理を検討しましょう。

3.2 フレームレート

• フレームレートの制限が設けられていません。
  • 必要に応じて、cap.set(cv2.CAP_PROP_FPS, fps)`を使用してフレームレートを設定するようにしましょう。

4. その他

• コード全体の可読性を向上させるために、コメントを追加することをおすすめします。
• コードフォーマットを統一すると、読みやすくなります。

5. 改善例

Python

import cv2
import configparser
import numpy as np

# INIファイルからRTSP URLを読み取る
config = configparser.ConfigParser()
config.read('streams.ini')

streams = [
    config.get('RTSP', 'stream1'),
    config.get('RTSP', 'stream2'),
    config.get('RTSP', 'stream3'),
    config.get('RTSP', 'stream4')
]

def get_frame_size(window_name):
    width = cv2.getWinProperty(window_name, cv2.WND_PROP_FRAME_WIDTH)
    height = cv2.getWinProperty(window_name, cv2.WND_PROP_FRAME_HEIGHT)
    return width, height

def resize_frame(frame, size):
    return cv2.resize(frame, size)

def combine_frames(frames, frame_width, frame_height):
    top_row = np.hstack((resize_frame(frames[0], (frame_width, frame_height)),
                         resize_frame(frames[1], (frame_width, frame_height))))
    bottom_row = np.hstack((resize_frame(frames[2], (frame_width, frame_height)),
                         resize_frame(frames[3], (frame_width, frame_height))))
    combined_frame = np.vstack((top_row, bottom_row))
    return combined_frame

# 各ストリームに対応するVideoCaptureオブジェクトを作成
caps = [cv2.VideoCapture(url) for url in streams]

for i, cap in enumerate(caps):
    if not cap.isOpened():
        print(f"Failed to open stream {i+1}: {streams[i]}")
        exit(1)

# ウィンドウを作成
cv2.namedWindow('RTSP Streams', cv2.WINDOW_NORMAL)

while True:
    frames = []