76 lines
2.1 KiB
Python
76 lines
2.1 KiB
Python
|
#!/usr/bin/env python
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
'''
|
|||
|
|
# @Time : 2024/10/11 10:43
|
|||
|
|
# @Author : hjw
|
|||
|
|
# @File : CameraHIK.py
|
|||
|
|
'''
|
|||
|
|
import cv2
|
|||
|
|
import socket
|
|||
|
|
|
|||
|
|
def portisopen(ip, port):
|
|||
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|||
|
|
sock.settimeout(1)
|
|||
|
|
state = sock.connect_ex((ip, port))
|
|||
|
|
if 0 == state:
|
|||
|
|
# print("port is open")
|
|||
|
|
return True
|
|||
|
|
else:
|
|||
|
|
# print("port is closed")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
class camera_HIK():
|
|||
|
|
|
|||
|
|
def __init__(self, ip, port, name, pw):
|
|||
|
|
# "rtsp://admin:zlzk.123@192.168.1.64:554"
|
|||
|
|
ret = portisopen(ip, port)
|
|||
|
|
self.camera_url = "rtsp://" + str(name) + ":" + str(pw) + "@" + str(ip) + ":" + str(port)
|
|||
|
|
self.ip = ip
|
|||
|
|
self.port = port
|
|||
|
|
self.init_success = False
|
|||
|
|
if ret:
|
|||
|
|
self.cap = cv2.VideoCapture(self.camera_url)
|
|||
|
|
self.init_success = True
|
|||
|
|
else:
|
|||
|
|
print('海康摄像头网络错误，请检测IP')
|
|||
|
|
|
|||
|
|
def get_img(self):
|
|||
|
|
ret = False
|
|||
|
|
frame = None
|
|||
|
|
if self.init_success==True:
|
|||
|
|
if portisopen(self.ip, self.port):
|
|||
|
|
# ret, frame = self.cap.read()
|
|||
|
|
# if ret == False:
|
|||
|
|
self.reconnect_camera()
|
|||
|
|
ret, frame = self.cap.read()
|
|||
|
|
else:
|
|||
|
|
print('海康摄像头网络断开')
|
|||
|
|
else:
|
|||
|
|
if portisopen(self.ip, self.port):
|
|||
|
|
self.reconnect_camera()
|
|||
|
|
ret, frame = self.cap.read()
|
|||
|
|
# if ret == False:
|
|||
|
|
# self.init_success =False
|
|||
|
|
# else:
|
|||
|
|
# self.init_success = True
|
|||
|
|
else:
|
|||
|
|
print('海康摄像头网络断开')
|
|||
|
|
|
|||
|
|
return ret, frame
|
|||
|
|
|
|||
|
|
def reconnect_camera(self):
|
|||
|
|
if self.init_success == True:
|
|||
|
|
self.cap.release()
|
|||
|
|
self.cap = cv2.VideoCapture(self.camera_url)
|
|||
|
|
ret, _ = self.cap.read()
|
|||
|
|
if ret:
|
|||
|
|
self.init_success = True
|
|||
|
|
else:
|
|||
|
|
self.init_success = False
|
|||
|
|
print("海康摄像头重连")
|
|||
|
|
|
|||
|
|
def release_camera(self):
|
|||
|
|
self.cap.release()
|
|||
|
|
|
|||
|
|
|