python opencv 图像数据 网络传输

主要目的:

利用工控机采集图像数据,再利用网络传输到笔记本电脑。在笔记本电脑上运行YOLOP模型进行freespace分割的推理,然后将分割结果返回到工控机,从而快速实现深度学校模型的部署测试。

客户端为工控机,服务器端为笔记本电脑

客户端:

import socket
import cv2
import numpy
import time
import sys
 
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

def SendVideo():
	#建立sock连接
	#address要连接的服务器IP地址和端口号
	address = ('192.168.14.97', 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.tobytes()
		
		#先发送要发送的数据的长度
		#ljust() 方法返回一个原字符串左对齐,并使用空格填充至指定长度的新字符串
		sock.send(str.encode(str(len(stringData)).ljust(16)))
		#发送数据
		sock.send(stringData)
		       
        #########接收服务器端处理的图像数据
		length = recvall(sock,16)#获得图片文件的长度,16代表获取长度
		SolvedstringData = recvall(sock, int(length))#根据获得的文件长度,获取图片文件
		solved_data = numpy.frombuffer(SolvedstringData, numpy.uint8)#将获取到的字符流数据转换成1维数组
		solved_img=cv2.imdecode(solved_data,cv2.IMREAD_COLOR)#将数组解码成图像

		cv2.imshow('CLIENT',frame)#显示客户端发送的图像

		cv2.imshow('solved_img',solved_img)#显示服务器端返回的图像
        #########


		#读取下一帧图片
		ret, frame = capture.read()
		if cv2.waitKey(10) == 27:
			break
	sock.close()
	
if __name__ == '__main__':
	SendVideo()

服务器端:

import socket
import time
import cv2
import numpy

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

def ReceiveVideo():
	#IP地址'0.0.0.0'为等待客户端连接
	address = ('192.168.14.235', 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)

	#接受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)#显示图像
		
		#进行下一步处理
		#。
		#。
		#。
        

        # ####  处理后的图像返回给客户端
		encode_param=[int (cv2.IMWRITE_JPEG_QUALITY),100]
		result, imgencode = cv2.imencode('.jpg', decimg, encode_param)
		#建立矩阵
		solved_data = numpy.array(imgencode)
		#将numpy矩阵转换成字符形式,以便在网络中传输
		stringSolved_data=solved_data.tobytes()
		#先发送要发送的数据的长度
		#ljust() 方法返回一个原字符串左对齐,并使用空格填充至指定长度的新字符串
		conn.send(str.encode(str(len(stringSolved_data)).ljust(16)))
		#发送数据
		conn.send(stringSolved_data)
        #####




		k = cv2.waitKey(10)&0xff
		if k == 27:
			break
	s.close()
	cv2.destroyAllWindows()
 
if __name__ == '__main__':
	ReceiveVideo()

 参考链接:

从同一网络下如何连接两台电脑?

家里是无线网 两台电脑可以同时上网 两台电脑的ip是一样的吗?-ZOL问答

python 实现TCP socket通信和 HTTP服务器、服务器和客户端通信python实例_算法技术博客的博客-CSDN博客

python - 通过 HTTP 发送和接收简单的 JSON 消息 - IT工具网

python json文件传输图片_chde2Wang的博客-CSDN博客

如何使用 Python 读取文件和照片的创建日期_python

如何用Python实现http客户端和服务器 - 萧海~ - 博客园

(python)【学习记录】http传输图片_hanbaoaaa的博客-CSDN博客_http传输图片

Http服务传输图片的Python实现_回眸郎的博客-CSDN博客_http图像的传递方式有哪些python

python利用opencv实现图像网络传输_树莓派poancv-其它代码类资源-CSDN下载

python opencv 图像网络传输_Chester Chai的博客-CSDN博客

python之socket编程 - 蒋璐 - 博客园

ros入门(四)--python实现话题订阅和发布及TCP数据转发 - 百度文库 

你可能感兴趣的:(机器人环境感知,网络,服务器,运维)