海康多相机同步取流保存图片

news/2024/7/9 23:59:20 标签: 深度学习, 目标检测

话不多说,直接上代码。代码里包含了多窗口显示图像和保存图片。

#"rtsp://admin:123456qq@192.168.10.192/stream1"
# rtsp://admin:Admin123@192.168.100.103:554/Streaming/Channels/101
#rtsp://admin:Admin123@192.168.100.103:554/cam/realmonitor?channel=1&subtype=0

import time
import multiprocessing as mp
import cv2
import os

"""
Source: Yonv1943 2018-06-17
https://github.com/Yonv1943/Python
https://zhuanlan.zhihu.com/p/38136322
OpenCV official demo
https://docs.opencv.org/3.0-beta/doc/py_tutorials/py_gui/py_video_display/py_video_display.html
海康、大华IpCamera RTSP地址和格式(原创,旧版)- 2014年08月12日 23:01:18 xiejiashu
rtsp_path_hikvison = "rtsp://%s:%s@%s/h265/ch%s/main/av_stream" % (user, pwd, ip, channel)
rtsp_path_dahua = "rtsp://%s:%s@%s/cam/realmonitor?channel=%d&subtype=0" % (user, pwd, ip, channel)
https://blog.csdn.net/xiejiashu/article/details/38523437
最新(2017)海康摄像机、NVR、流媒体服务器、回放取流RTSP地址规则说明 - 2017年05月13日 10:51:46 xiejiashu
rtsp_path_hikvison = "rtsp://%s:%s@%s//Streaming/Channels/%d" % (user, pwd, ip, channel)
https://blog.csdn.net/xiejiashu/article/details/71786187
"""


def image_put(q, user, pwd, ip, channel=1):
    # cap = cv2.VideoCapture("rtsp://%s:%s@%s/stream1" % (user, pwd, ip))
    # cap = cv2.VideoCapture("rtsp://%s:%s@%sStreaming/Channels/1701?transportmode=unicast" % (user, pwd, ip))

    # if cap.isOpened():
    #     print('HIKVISION')
    # else:
    #     cap = cv2.VideoCapture("rtsp://%s:%s@%s/cam/realmonitor?channel=%d&subtype=0" % (user, pwd, ip, channel))
    #     print('DaHua')
    cap = cv2.VideoCapture("rtsp://%s:%s@%s/cam/realmonitor?channel=%d&subtype=0" % (user, pwd, ip, channel))
    if cap.isOpened():
        print('HIKVISION')
    else:
        cap = cv2.VideoCapture("rtsp://admin:123456qq@192.168.100.12/stream1")
        print('Genius')

    while True:
        q.put(cap.read()[1])
        q.get() if q.qsize() > 1 else time.sleep(0.01)



def run_opencv_camera():
    user, pwd, ip, channel = "admin", "123456qq", "33.90.7.110", 1

    cap_path = 0  # local camera (e.g. the front camera of laptop)
    # cap_path = 'video.avi'  # the path of video file
    # cap_path = "rtsp://%s:%s@%s/h264/ch%s/main/av_stream" % (user, pwd, ip, channel)  # HIKIVISION old version 2015
    # cap_path = "rtsp://%s:%s@%s//Streaming/Channels/%d" % (user, pwd, ip, channel)  # HIKIVISION new version 2017
    # cap_path = "rtsp://%s:%s@%s/cam/realmonitor?channel=%d&subtype=0" % (user, pwd, ip, channel)  # dahua

    cap = cv2.VideoCapture(cap_path)

    while cap.isOpened():
        is_opened, frame = cap.read()
        cv2.imshow('frame', frame)
        cv2.waitKey(1000)
    cap.release()

def image_get_s(queue_list, ip, img_path):
    """show in single opencv-imshow window"""

    window_name1 = "%s" % ip[0]
    window_name2 = "%s" % ip[1]
    window_name3 = "%s" % ip[2]
    # window_name4 = "%s" % ip[3]

    idx_img = 0
    cv2.namedWindow(window_name1, flags=cv2.WINDOW_FREERATIO)
    cv2.namedWindow(window_name2, flags=cv2.WINDOW_FREERATIO)
    cv2.namedWindow(window_name3, flags=cv2.WINDOW_FREERATIO)
    # cv2.namedWindow(window_name4, flags=cv2.WINDOW_FREERATIO)
    while True:
        imgs = [q.get() for q in queue_list]
        img1 = imgs[0]
        img2 = imgs[1]
        img3 = imgs[2]
        # img4 = imgs[3]

        img1_tmp = cv2.resize(img1, (int(1920/2 ), int(1080/2 )))
        img2_tmp = cv2.resize(img2, (int(1920/2 ), int(1080/2 )))
        img3_tmp = cv2.resize(img3, (int(1920 / 2), int(1080 / 2)))
        # img4_tmp = cv2.resize(img4, (int(1920 / 2), int(1080 / 2)))

        cv2.imshow(window_name1, img1_tmp)
        cv2.imshow(window_name2, img2_tmp)
        cv2.imshow(window_name3, img3_tmp)
        # cv2.imshow(window_name4, img4_tmp)
        # imgs = np.concatenate(imgs, axis=1)
        # cv2.imshow(window_name, imgs)
        key = cv2.waitKey(1)

        if key == ord('s'):
            img_path0 = os.path.join(img_path[0], str(idx_img) + '.jpg')
            cv2.imwrite(img_path0, img1)

            img_path1 = os.path.join(img_path[1], str(idx_img) + '.jpg')
            cv2.imwrite(img_path1, img2)

            img_path2 = os.path.join(img_path[2], str(idx_img) + '.jpg')
            cv2.imwrite(img_path2, img3)

            # img_path3 = os.path.join(img_path[3], str(idx_img) + '.jpg')
            # cv2.imwrite(img_path3, img3)
            # idx_img += 1

def image_get_all(queue_list, ip, img_path):
    """show in single opencv-imshow window"""

    window_name1 = "%s" % ip[0]
    window_name2 = "%s" % ip[1]
    # window_name3 = "%s" % ip[2]
    # window_name4 = "%s" % ip[3]


    idx_img = 0
    cv2.namedWindow(window_name1, flags=cv2.WINDOW_FREERATIO)
    cv2.namedWindow(window_name2, flags=cv2.WINDOW_FREERATIO)
    # cv2.namedWindow(window_name3, flags=cv2.WINDOW_FREERATIO)
    # cv2.namedWindow(window_name4, flags=cv2.WINDOW_FREERATIO)
    while True:
        imgs = [q.get() for q in queue_list]
        img1 = imgs[0]
        img2 = imgs[1]
        # img3 = imgs[2]
        # img4 = imgs[3]

        img1_tmp = cv2.resize(img1, (int(1920/2 ), int(1080/2 )))
        img2_tmp = cv2.resize(img2, (int(1920/2 ), int(1080/2 )))
        # img3_tmp = cv2.resize(img3, (int(1920/4), int(1080/ 4)))
        # img4_tmp = cv2.resize(img4, (int(1920 / 4), int(1080 / 4)))
        # print('11',len(img1_tmp))
        cv2.imshow(window_name1, img1_tmp)
        cv2.imshow(window_name2, img2_tmp)
        # cv2.imshow(window_name3, img3_tmp)
        # cv2.imshow(window_name4, img4_tmp)
        # imgs = np.concatenate(imgs, axis=1)
        # cv2.imshow(window_name, imgs)
        cv2.waitKey(1)

        cv2.imwrite(img_path[0]+str(idx_img)+'.jpg', img1)
        cv2.imwrite(img_path[1]+str(idx_img)+'.jpg', img2)
        # cv2.imwrite(img_path[2]+str(idx_img)+'.jpg', img3)
        # cv2.imwrite(img_path[3]+str(idx_img) +'.jpg',img4)
        idx_img +=1


def run_single_camera():
    # user_name, user_pwd, camera_ip = "admin", "admin123456", "172.20.114.196"
    # user_name, user_pwd, camera_ip = "admin", "Admin123", "192.168.100.103:554"
    user_name, user_pwd, camera_ip = "admin", "Admin123", "192.168.100.100:554"


    mp.set_start_method(method='spawn')  # init
    queue = mp.Queue(maxsize=2)
    processes = [mp.Process(target=image_put, args=(queue, user_name, user_pwd, camera_ip)),
                 mp.Process(target=image_get_s, args=(queue, camera_ip))]

    [process.start() for process in processes]
    [process.join() for process in processes]


def run_multi_camera():
    user_name, user_pwd = "admin", "Admin123"

    camera_ip_l = [
        # "172.20.114.196",  # ipv4
        # "[fe80::3aaf:29ff:fed3:d260]",  # ipv6
        # "192.168.100.180:554",
        # "192.168.100.182:554",
        "192.168.100.181:554",
        "192.168.100.184:554",
    ]

    mp.set_start_method(method='spawn')  # init
    queues = [mp.Queue(maxsize=len(camera_ip_l)) for _ in camera_ip_l]


    imgs_path = []
    for i in range(len(camera_ip_l)):
        imgs_path.append("E:/dataset_0902/images/{}/".format(camera_ip_l[i].split('.')[-1].split(':')[0]))
        if not os.path.exists(imgs_path[-1]):
            os.makedirs(imgs_path[-1])

    processes = [mp.Process(target=image_get_all, args=(queues, camera_ip_l, imgs_path))]

    for queue, camera_ip, img_path in zip(queues, camera_ip_l, imgs_path):
        processes.append(mp.Process(target=image_put, args=(queue, user_name, user_pwd, camera_ip)))



    for process in processes:
        process.daemon = True
        process.start()
    for process in processes:
        process.join()
    # p.join()

def image_collect(queue_list, camera_ip_l):
    import numpy as np

    """show in single opencv-imshow window"""
    window_name1 = "%s_and_so_no" % camera_ip_l[0]
    window_name2 = "%s_and_so_no" % camera_ip_l[1]
    cv2.namedWindow(window_name1, flags=cv2.WINDOW_FREERATIO)
    cv2.namedWindow(window_name2, flags=cv2.WINDOW_FREERATIO)
    while True:
        imgs = [q.get() for q in queue_list]
        img1 = imgs[0]
        img2 = imgs[1]

        # cv2.imshow(window_name1, img1)
        # cv2.imshow(window_name2, img2)

        img1_tmp = cv2.resize(img1, (int(1920 / 2), int(1080 / 2)))
        img2_tmp = cv2.resize(img2, (int(1920 / 2), int(1080 / 2)))
        cv2.imshow(window_name1, img1_tmp)
        cv2.imshow(window_name2, img2_tmp)
        # imgs = np.concatenate(imgs, axis=1)
        # cv2.imshow(window_name, imgs)
        cv2.waitKey(1)

        # if key == ord('s'):
        #     img_path = os.path.join(img_folder, str(idx_img) + '.jpg')
        #     cv2.imwrite(img_path, img)
        #     idx_img += 1
    # """show in multiple opencv-imshow windows"""
    # [cv2.namedWindow(window_name, flags=cv2.WINDOW_FREERATIO)
    #  for window_name in camera_ip_l]
    # while True:
    #     for window_name, q in zip(camera_ip_l, queue_list):
    #         cv2.imshow(window_name, q.get())
    #         cv2.waitKey(1)


def run_multi_camera_in_a_window():
    user_name, user_pwd = "admin", "Admin123"
    camera_ip_l = [
        # "172.20.114.196",  # ipv4
        # "[fe80::3aaf:29ff:fed3:d260]",  # ipv6
        "192.168.100.182:554",
        "192.168.100.183:554"
    ]

    mp.set_start_method(method='spawn')  # init
    queues = [mp.Queue(maxsize=2) for _ in camera_ip_l]

    processes = [mp.Process(target=image_collect, args=(queues, camera_ip_l))]
    for queue, camera_ip in zip(queues, camera_ip_l):
        processes.append(mp.Process(target=image_put, args=(queue, user_name, user_pwd, camera_ip)))

    for process in processes:
        process.daemon = True  # setattr(process, 'deamon', True)
        process.start()
    for process in processes:
        process.join()



def get_biaoding():
    user, pwd, ip, channel = "admin", "Admin123", "192.168.100.182:554", 1
    cap = cv2.VideoCapture("rtsp://%s:%s@%s/cam/realmonitor?channel=%d&subtype=0" % (user, pwd, ip, channel))
    if cap.isOpened():
        print('HIKVISION')
    idx_img = 20
    img_folder = "D:/20210226/camera{}".format(ip.split('.')[-1].split(':')[0])
    if not os.path.exists(img_folder):
        os.makedirs(img_folder)
    while True:
        img = cap.read()[1]
        img_tmp = cv2.resize(img, (1920 // 2, 1080 // 2))
        cv2.imshow('image', img_tmp)
        key = cv2.waitKey(1)
        if key == ord('s'):
            img_path = os.path.join(img_folder, str(idx_img) + '.jpg')
            cv2.imwrite(img_path, img)
            idx_img += 1




def get_img():
    cap1 = cv2.VideoCapture("rtsp://%s:%s@%s/cam/realmonitor?channel=%d&subtype=0" % ("admin", "Admin123", "192.168.100.181:554/", 1))
    # cap2 = cv2.VideoCapture("rtsp://%s:%s@%s/cam/realmonitor?channel=%d&subtype=0" % ("admin", "Admin123", "192.168.100.182:554/", 1))
    # cap3 = cv2.VideoCapture("rtsp://%s:%s@%s/cam/realmonitor?channel=%d&subtype=0" % ("admin", "Admin123", "192.168.100.183:554/", 1))
    # cap4 = cv2.VideoCapture("rtsp://%s:%s@%s/cam/realmonitor?channel=%d&subtype=0" % ("admin", "Admin123", "192.168.100.184:554/", 1))

    img_save1 = "E:/dataset_0902/calibrate_intri/181"
    # img_save1 = "D:/20210308/camera/181"
    # img_save2 = "D:/20210225/camera/182"
    # img_save3 = "D:/20210225/camera/183"
    # img_save4 = "D:/20210225/camera/184"
    if not os.path.exists(img_save1):
        os.makedirs(img_save1)
    # if not os.path.exists(img_save2):
    #     os.makedirs(img_save2)
    # if not os.path.exists(img_save3):
    #     os.makedirs(img_save3)
    # if not os.path.exists(img_save4):
    #     os.makedirs(img_save4)

    if cap1.isOpened():
        print('HIKVISION')

    idx_img = 0
    while True:

        time1 = cap1.get(cv2.CAP_PROP_POS_MSEC)

        print(time1)

        img1 = cap1.read()[1]
        if (img1 is None):
            print("ok")
            continue

        cv2.imwrite(img_save1 + '/' + str(idx_img) + '.jpg', img1)
        idx_img = idx_img + 1


        # key = cv2.waitKey(1)
        # if key == ord('s'):
        #     cv2.imwrite()
        #     cv2.imwrite(iimg_save1 + '/' + str(idx_img) + '.jpg', img1mg_save2 + '/' + str(idx_img) + '.jpg', img2)
        #     cv2.imwrite(img_save3 + '/' + str(idx_img) + '.jpg', img3)
        #     cv2.imwrite(img_save4 + '/' + str(idx_img) + '.jpg', img4)
        #     idx_img += 1

        img1 = cv2.resize(img1, (int(1920 / 4), int(1080 / 4)))

        cv2.imshow('1', img1)
        cv2.waitKey(1)


if __name__ == '__main__':

    # get_biaoding()
    # get_img()

    run_multi_camera()

http://www.niftyadmin.cn/n/5142488.html

相关文章

解决问题Conda:CondaValueError: Malformed version string ‘~’ : invalid character(s)

解决问题Conda:CondaValueError: Malformed version string ‘~’ : invalid character(s) 背景 今天使用Conda构建项目运行环境的时候报错::CondaValueError: Malformed version string ‘~’ : invalid character(s) ##报错问题 在安装te…

设计模式——模板模式(Template Pattern)+ Spring相关源码

文章目录 一、模板模式定义二、例子2.1 菜鸟教程例子2.1.1 抽象类Game 定义了play方法的执行步骤。2.1.2 继承Game类并实现initialize、startPlay、endPlay方法。2.1.3 使用 2.2 JDK源码 —— Map 2.3 Spring源码 —— JdbcTemplate2.4 Spring源码 —— RestTemplate三、其他设…

nodejs+vue+elementui+python家电销售分析系统设计与实现-计算机毕业设计

系统按照用户的实际需求开发而来,贴近生活。从管理员通过正确的账号的密码进入系统,可以使用相关的系统应用。管理员总体负责整体系统的运行维护,统筹协调。 我们可以利用计算机技术来取代传统的管理模式,实现家电销售分析系统的…

〖Python网络爬虫实战㊱〗- JavaScript 网站加密和混淆

订阅:新手可以订阅我的其他专栏。免费阶段订阅量1000+python项目实战 Python编程基础教程系列(零基础小白搬砖逆袭) 说明:本专栏持续更新中,订阅本专栏前必读关于专栏〖Python网络爬虫实战〗转为付费专栏的订阅说明作者:爱吃饼干的小白鼠。Python领域优质创作者,2022年度…

windows docker desktop 更换镜像 加速

最近 docker hub 访问不了; 经过研究 可以通过添加 代理镜像网址 添加代理服务器的方式 实现完美访问 1添加镜像网站 修改成国内镜像地址就能享受到飞一般的速度,但有一个问题,部分站点镜像不全或者镜像比较老,建议使用多个镜像站。 https…

vue图书馆书目推荐数据分析与可视化-计算机毕业设计python-django-php

建立本图书馆书目推荐数据分析是为了通过系统对图书数据根据算法进行的分析好推荐,以方便用户对自己所需图书信息的查询,根据不同的算法机制推荐给不同用户不同的图书,用户便可以从系统中获得图书信息信息。 对用户相关数据进行分析&#xff…

CSS标点符号换行问题

最近遇到一个奇怪的现象,元素中中文文本正常显示,但是加了一堆符号后中文文本居然换行了. div{width: 200px;border: 1px solid blue;word-break: break-all;} <div>文本</div>经过研究发现&#xff0c;因为标点符号不允许出现在行首和行尾&#xff0c;连带着符号…

C++——list

目录 list介绍 list的函数接口 构造函数 push_front和pop_front push_back和pop_back insert erase 迭代器 front和back size resize empty clear list::sort unique reverse 迭代器的实现 list介绍 list是一种可以在常数范围内在任意位置进行插入和删除的序列…