python openCV 基於TCP的socket網絡傳輸視頻（一）

create time:2019年7月17日

  我在網上找了好多，現在基本實現了傳輸視頻的功能，其思路是：首先得先了解如何使用openCV採集攝像頭數據，
其次得了解python socket網絡編程。因爲socket不能直接傳輸openCV採集的數據，所以還需要轉碼操作。
  在下個版本，我用的類的方式實現了視頻傳輸，這個版本先用平常方法實現的，也是一個客戶端，一個服務端。

服務端

#!/usr/bin/python
# -*-coding:utf-8 -*-

import socket
import cv2
import numpy

# 接受圖片大小的信息
def recv_size(sock, count):
     buf = ''
     while count:
         newbuf = sock.recv(count)
         if not newbuf: return None
         buf += newbuf
         count -= len(newbuf)
     return buf

# socket.AF_INET 用於服務器與服務器之間的網絡通信
#socket.SOCK_STREAM 代表基於TCP的流式socket通信
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 設置地址與端口，如果是接收任意ip對本服務器的連接，地址欄可空，但端口必須設置
address = ('10.0.0.30', 8010)
s.bind(address)  # 將Socket（套接字）綁定到地址
s.listen(True)   # 開始監聽TCP傳入連接
print ('Waiting for images...')

# 接受TCP鏈接並返回（conn, addr），其中conn是新的套接字對象，可以用來接收和發送數據，addr是鏈接客戶端的地址。

conn, addr = s.accept()
while True:
     length = recv_size(conn,16)  # 首先接收來自客戶端發送的大小信息
     if isinstance(length, str):   # 若成功接收到大小信息，進一步再接收整張圖片
         stringData = recv_size(conn,int(length))
         data =numpy.fromstring(stringData, dtype='uint8')
         decimg = cv2.imdecode(data,1)  # 解碼處理，返回mat圖片
         cv2.imshow('SERVER', decimg)
         if cv2.waitKey(10) == 27:
             break
         print('Image recieved successfully!')
     if cv2.waitKey(10) == 27:
         break
s.close()
cv2.destroyAllWindows()

客戶端

#!/usr/bin/python
# -*-coding:utf-8 -*-
import socket
import cv2
import numpy
from time import sleep

# socket.AF_INET 用於服務器與服務器之間的網絡通信
# socket.SOCK_STREAM 代表基於TCP的流式socket通信
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 連接服務端
address_server = ('10.0.0.30', 8888)
sock.connect(address_server)

# 從攝像頭採集圖像
# 參數是0,表示打開筆記本的內置攝像頭,參數是視頻文件路徑則打開視頻
capture = cv2.VideoCapture(0)
# capture.read() 按幀讀取視頻
# ret,frame 是capture.read()方法的返回值
# 其中ret是布爾值，如果讀取幀正確，返回True;如果文件讀到末尾，返回False。
# frame 就是每一幀圖像，是個三維矩陣
ret, frame = capture.read()
encode_param = [int(cv2.IMWRITE_JPEG_QUALITY), 50]
while ret:
    # 首先對圖片進行編碼，因爲socket不支持直接發送圖片
    # '.jpg'表示把當前圖片frame按照jpg格式編碼
    # result, img_encode = cv2.imencode('.jpg', frame)
    img_encode = cv2.imencode('.jpg', frame, encode_param)[1]
    data = numpy.array(img_encode)
    stringData = data.tostring()
    # 首先發送圖片編碼後的長度
    sock.send(str(len(stringData)).ljust(16))
    # 然後一個字節一個字節發送編碼的內容
    # 如果是python對python那麼可以一次性發送，如果發給c++的server則必須分開發因爲編碼裏面有字符串結束標誌位，c++會截斷
    # for i in range(0, len(stringData)):
    #     sock.send(stringData[i])
    sock.send(stringData)
    # sleep(1)
    ret, frame = capture.read()
    cv2.resize(frame, (640, 480))

sock.close()
cv2.destroyAllWindows()