python 利用opencv实现图像网络传输

举报
泽宇-Li 发表于 2020/11/13 08:50:37 2020/11/13
【摘要】 本代码主要实现的是利用网络传输图片,用在我的树莓派项目之上。该项目在PC上运行服务端,树莓派上运行客户端,两者连接到同一局域网中,修改代码中的IP地址,就可以实现将树莓派采集到的图像数据实时传输到PC端。先运行服务端代码,然后运行客户端代码即可。树莓派摄像头使用的是普通的USB摄像头,并且在树莓派上安装了opencv,在树莓派上安装opencv的过程可以参考https://www.pyima...

本代码主要实现的是利用网络传输图片,用在我的树莓派项目之上。该项目在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地址和端口号。

服务端代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
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()

客户端代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
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()



【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

举报
请填写举报理由
0/200