python 利用opencv實現影象網路傳輸
阿新 • • 發佈:2020-11-15
本程式碼主要實現的是利用網路傳輸圖片,用在我的樹莓派專案之上。該專案在PC上執行服務端,樹莓派上執行客戶端,兩者連線到同一區域網中,修改程式碼中的IP地址,就可以實現將樹莓派採集到的影象資料實時傳輸到PC端。先執行服務端程式碼,然後執行客戶端程式碼即可。樹莓派攝像頭使用的是普通的USB攝像頭,並且在樹莓派上安裝了opencv,在樹莓派上安裝opencv的過程可以參考https://www.pyimagesearch.com/2017/09/04/raspbian-stretch-install-opencv-3-python-on-your-raspberry-pi/。最後,該程式碼稍加修改就可以傳輸其他的資訊,當然服務端,客戶端也可以同時在PC上執行,以驗證結果。所以本質還是希望讀者藉此程式碼可以瞭解python的socket程式設計。程式碼意義已在註釋中詳細說明,僅供參考。使用時請注意修改IP地址和埠號。
服務端程式碼
import socket import time import cv2 import numpy def ReceiveVideo(): #IP地址'0.0.0.0'為等待客戶端連線 address = ('0.0.0.0',8002) #建立socket物件,引數意義見https://blog.csdn.net/rebelqsp/article/details/22109925 #socket.AF_INET:伺服器之間網路通訊 #socket.SOCK_STREAM:流式socket,for TCP s = socket.socket(socket.AF_INET,socket.SOCK_STREAM) #將套接字繫結到地址,在AF_INET下,以元組(host,port)的形式表示地址. s.bind(address) #開始監聽TCP傳入連線。引數指定在拒絕連線之前,作業系統可以掛起的最大連線數量。該值至少為1,大部分應用程式設為5就可以了。 s.listen(1) def recvall(sock,count): buf = b''#buf是一個byte型別 while count: #接受TCP套接字的資料。資料以字串形式返回,count指定要接收的最大資料量. newbuf = sock.recv(count) if not newbuf: return None buf += newbuf count -= len(newbuf) return buf #接受TCP連線並返回(conn,address),其中conn是新的套接字物件,可以用來接收和傳送資料。addr是連線客戶端的地址。 #沒有連線則等待有連線 conn,addr = s.accept() print('connect from:'+str(addr)) while 1: start = time.time()#用於計算幀率資訊 length = recvall(conn,16)#獲得圖片檔案的長度,16代表獲取長度 stringData = recvall(conn,int(length))#根據獲得的檔案長度,獲取圖片檔案 data = numpy.frombuffer(stringData,numpy.uint8)#將獲取到的字元流資料轉換成1維陣列 decimg=cv2.imdecode(data,cv2.IMREAD_COLOR)#將陣列解碼成影象 cv2.imshow('SERVER',decimg)#顯示影象 #進行下一步處理 #。 #。 #。 #將幀率資訊回傳,主要目的是測試可以雙向通訊 end = time.time() seconds = end - start fps = 1/seconds; conn.send(bytes(str(int(fps)),encoding='utf-8')) k = cv2.waitKey(10)&0xff if k == 27: break s.close() cv2.destroyAllWindows() if __name__ == '__main__': ReceiveVideo()
客戶端程式碼:
import socket import cv2 import numpy import time def SendVideo(): #建立sock連線 #address要連線的伺服器IP地址和埠號 address = ('127.0.0.1',8002) try: #建立socket物件,引數意義見https://blog.csdn.net/rebelqsp/article/details/22109925 #socket.AF_INET:伺服器之間網路通訊 #socket.SOCK_STREAM:流式socket,for TCP sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM) #開啟連線 sock.connect(address) except socket.error as msg: print(msg) sys.exit(1) #建立影象讀取物件 capture = cv2.VideoCapture(0) #讀取一幀影象,讀取成功:ret=1 frame=讀取到的一幀影象;讀取失敗:ret=0 ret,frame = capture.read() #壓縮引數,後面cv2.imencode將會用到,對於jpeg來說,15代表影象質量,越高代表影象質量越好為 0-100,預設95 encode_param=[int(cv2.IMWRITE_JPEG_QUALITY),15] while ret: #停止0.1S 防止傳送過快服務的處理不過來,如果服務端的處理很多,那麼應該加大這個值 time.sleep(0.01) #cv2.imencode將圖片格式轉換(編碼)成流資料,賦值到記憶體快取中;主要用於影象資料格式的壓縮,方便網路傳輸 #'.jpg'表示將圖片按照jpg格式編碼。 result,imgencode = cv2.imencode('.jpg',frame,encode_param) #建立矩陣 data = numpy.array(imgencode) #將numpy矩陣轉換成字元形式,以便在網路中傳輸 stringData = data.tostring() #先發送要傳送的資料的長度 #ljust() 方法返回一個原字串左對齊,並使用空格填充至指定長度的新字串 sock.send(str.encode(str(len(stringData)).ljust(16))); #傳送資料 sock.send(stringData); #讀取伺服器返回值 receive = sock.recv(1024) if len(receive):print(str(receive,encoding='utf-8')) #讀取下一幀圖片 ret,frame = capture.read() if cv2.waitKey(10) == 27: break sock.close() if __name__ == '__main__': SendVideo()
以上就是python 利用opencv實現影象網路傳輸的詳細內容,更多關於python 影象網路傳輸的資料請關注我們其它相關文章!