ValueError: cannot reshape array of size 65495 into shape (480,640,3)の解決法

Question

前提・実現したいこと

cv2とsocketでwebカメラから撮った動画をライブ中継したい。

発生している問題・エラーメッセージ

###ValueError: cannot reshape array of size 65495 into shape (480,640,3)
普段使っているときは問題ないのですが、カメラに指を近づけたり、急に離したりするとエラーが出ます。
あと、sizeはその時ごとに変わります。

該当のソースコード

###server.py

import socket
import numpy as np
import cv2
import time

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)#ソケットオブジェクト作成
s.bind(("192.168.0.14", 6666)) # サーバー側PCのipと使用するポート
print("接続待機中")
s.listen(1) # 接続要求を待機
soc, addr = s.accept() # 要求が来るまでブロック
print(str(addr)+"と接続完了")

cam = cv2.VideoCapture(0)#カメラオブジェクト作成

while (True):

flag,img = cam.read()       #カメラから画像データを受け取る
img = img.tostring()        #numpy行列からバイトデータに変換
soc.send(img)              # ソケットにデータを送信
k = cv2.waitKey(1)         #↖ 
if k== 13 :                #←　ENTERキーで終了
    break                  #↙

cam.releace() #カメラオブジェクト破棄

###client.py

import socket
import numpy as np
import cv2

soc = socket.socket(socket.AF_INET, socket.SOCK_STREAM)#ソケットオブジェクト作成

soc.connect(("192.168.0.14", 6666))#サーバー側のipと使用するポート(ポートはサーバーと同じにする。)

print("接続完了")

while(1):
data = soc.recv(921600)
data = np.frombuffer(data,dtype=np.uint8)#バイトデータ→ndarray変換
data = np.reshape(data,(480,640,3))#形状復元
cv2.imshow("",data);
k = cv2.waitKey(1)
if k== 13 :
break

cv2.destroyAllWindows() # 作成したウィンドウを破棄

Answer 1

ソケットの利用

send と recv はネットワークバッファに働きかけるものだ。だから、手渡したもの (や返してもらいたいもの) を 1 バイトも残さず実際に処理してくれているとは限らない。一般的に言って、 send はバッファが埋まるまで、recv はバッファが空になるまで処理をして、そのバイト数を返す。メッセージが完全に処理されるまで繰り返し呼び出すのは 自分の 責任なのだ。

とあるようにrecvでサイズ指定しても、必ずそのサイズだけ返してくれる保証はありません。
バッファが一杯なら、途中までのデータしか返してくれない可能性があります。
その場合、欲しいサイズに達するまで繰り返しrecvする必要があります。

とりあえず、お互いワンセットとして処理したいメッセージサイズが固定(921600)なのであれば、client.pyを以下のように修正することで実現できます。

# 略

MSGLEN = 921600
while(1):

    #data = soc.recv(921600)

    # 指定サイズ受信完了するまで繰り返し受信する
    chunks = []
    bytes_recd = 0
    while bytes_recd < MSGLEN:
        chunk = soc.recv(min(MSGLEN - bytes_recd, MSGLEN)) # １回ですべて受信できなかったら残り（少ない方）を読み込む
        if chunk == b'':
            raise RuntimeError("socket connection broken")
        chunks.append(chunk)
        bytes_recd = bytes_recd + len(chunk)

    data = b''.join(chunks) # 分割して受信されたデータを結合

    data = np.frombuffer(data,dtype=np.uint8)#バイトデータ→ndarray変換
    
    # 略

Comments

[hoge1243]

すみません、client.pyをそのように直したのですが、エラーは出ないのですが一向に処理が進みません。
一体どうしてでしょうか？

[can110]

「処理が進まない」という状況が分かりませんが、当方環境では進みました。