1. 程式人生 > 程式設計 >python 利用opencv實現影象網路傳輸

python 利用opencv實現影象網路傳輸

本程式碼主要實現的是利用網路傳輸圖片,用在我的樹莓派專案之上。該專案在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 影象網路傳輸的資料請關注我們其它相關文章!