基于TCP网络传输协议和python3 的双路视频数据和单路指令传输显示系统的代码

注意:这个代码是很早之前的写的一个小代码,可能会有一些地方会出现一些小问题,但是我在局域网的情况下已经测试过了。仅仅是局域网

 服务端代码:

#!/usr/bin/env python
# -*- coding:utf-8 -*- 
# Author: K_liu
'''
这是一个服务器端程序
可以通过TCP传输协议实现两路视频传输,一路指令传输,本程序采用一个随机数来模拟指令
'''
import socket
import struct
import time
import cv2
import numpy
import random

# 设置套接字,本程序采用TCP协议,当然也可以采用UDP协议,但是要保证指令传输的正确性,所以采用了TCP协议。
addr = ("0.0.0.0", 8080)
server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server_socket.setsockopt(socket.SOL_SOCKET,socket.SO_REUSEADDR, 1) #端口可复用
server_socket.bind(addr) # 绑定地址端口
server_socket.listen(True) # 进入监听状态

def Package(camera,img, img_resolution, img_param, conn_client, flag):# 注意flag是一个数据区分标志,0和1分别表示两路数据
    '本函数的主要功能是实现Opencv采集到的视频数据进行转码,同时进行数据打包传输'
    number = random.randint(1,10) # 生成一个1到10的随机数,模拟指令
    frame = cv2.resize(img, img_resolution) # 对图像数据大小进行设定
    _, frame_encode = cv2.imencode('.jpg', frame, img_param) # 进行编码,编码成可以传输的jpg格式,Opencv的数据是不能直接传输的
    numpy_frame = numpy.array(frame_encode) # 将编码的图片转成numpy的数组
    data_frame = numpy_frame.tostring() # 转成字符串,网络传输只能传输str格式
    try:
        # 按照相应的格式进行打包发送图片
        conn_client.send(struct.pack("lhh", len(numpy_frame), flag, number)
                         + data_frame)
    except:
        camera.release()
        return 0

def Send_image(conn_client):
    img_fps = 30 # 设置帧率
    img_resolution = (640, 480)
    # 初始化摄像头
    camera1 = cv2.VideoCapture(0)
    camera2 = cv2.VideoCapture(1)
    img_param = [int(cv2.IMWRITE_JPEG_QUALITY), img_fps]  # 设置传送图像格式、帧数
    while(1): # 下面的就不用注释了吧
        time.sleep(0.1)
        ret1, frame1 = camera1.read()
        ret2, frame2 = camera2.read()
        Package(camera1,frame1,img_resolution,img_param,conn_client,0)
        time.sleep(0.1)
        Package(camera1, frame2, img_resolution, img_param, conn_client, 1)

if __name__ == '__main__':
    print('Waiting the connecting ...')
    conn, addr = server_socket.accept()
    print('connected form the: {}'.format(addr))
    Send_image(conn)


 客户端代码

#!/usr/bin/env python
# -*- coding:utf-8 -*- 
# Author: K_liu
'''
这是一个客户端程序,由于需要显示两路视频画面,所以采用pygame来进行编写程序,Opencv界面可定制化没有pygame灵活
当然大家也可以采用python的其他GUI开发工具。
'''
import socket
import cv2
import struct
import numpy
import pygame
from pygame.locals import *
from sys import exit
global image1
global image2

# Configuration file
pygame.init() # 初始化pygame
window_size = (1280, 480) # 设置显示界面大小
screen = pygame.display.set_mode(window_size, 0, 32) # pygame显示设置
pygame.display.set_caption('Two Camera Show') # 设置显示标题
pygame.display.flip()
clock = pygame.time.Clock()

def Get_image():
    info = struct.unpack("lhh", client_socket.recv(8)) # 进行数据包解包操作
    flag = info[1] # flag是为了辨别是那一路信号,采用0和1进行区分,详情请见服务器段程序
    print('模拟指令:{}'.format(info[2]))
    buf_size = info[0]  # 获取读的图片总长度
    if buf_size:
        try:
            buf = b""  # 代表bytes类型
            temp_buf = buf
            while (buf_size):  # 读取每一张图片的长度
                temp_buf = client_socket.recv(buf_size)
                buf_size -= len(temp_buf)
                buf += temp_buf  # 获取图片
                data = numpy.fromstring(buf, dtype='uint8')  # 按uint8转换为图像矩阵
                image = cv2.imdecode(data, 1)  # 图像解码
                return image, flag
        except:
            pass

def ShowImage():
    while (1):
        for event in pygame.event.get():
            if event.type == QUIT:
                exit()
        frame, flag = Get_image()
        if flag == 0:
            image1 = frame
            flag1 = flag
            frame1 = cv2.cvtColor(image1, cv2.COLOR_BGR2RGB)#将图片转换成RGB格式,Opencv采用的是BGR
            frame1 = numpy.rot90(frame1)
            frame1 = cv2.flip(frame1, 0, dst= None)
            frame1 = pygame.surfarray.make_surface(frame1) # 转换成pygame的背景格式
            screen.blit(frame1, (0, 0)) # 进行背景渲染

        elif flag == 1:
            image2 = frame
            flag2 = flag
            frame2 = cv2.cvtColor(image2, cv2.COLOR_BGR2RGB)
            frame2 = numpy.rot90(frame2)
            frame2 = pygame.surfarray.make_surface(frame2)
            screen.blit(frame2, (640,0))
        pygame.display.update() # 刷新背景



if __name__ == '__main__':
    addr_port = ('127.0.0.1', 8080)
    client_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    client_socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)  # 端口可复用
    client_socket.connect(addr_port)
    print("IP is %s:%d" % (addr_port[0], addr_port[1]))
    ShowImage()


 

发布了15 篇原创文章 · 获赞 22 · 访问量 2万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览