231 lines
8.6 KiB
Python
Executable File
231 lines
8.6 KiB
Python
Executable File
# -*- coding:utf-8 -*-
|
|
|
|
import cv2
|
|
import time
|
|
import redis
|
|
import queue
|
|
import sys
|
|
import time
|
|
import threading
|
|
from System_I import Check_start_fordataselect,Density,Detection_Y5,Fine_grained
|
|
from common import global_para as gp
|
|
r_db = None
|
|
|
|
|
|
import logging
|
|
from logging.handlers import TimedRotatingFileHandler
|
|
import os,torch
|
|
|
|
|
|
class cusLogger:
|
|
def __init__(self, loggerName):
|
|
|
|
# 创建一个logger
|
|
self.logger = logging.getLogger(loggerName)
|
|
self.logger.setLevel(logging.DEBUG)
|
|
# 创建一个handler，用于写入日志文件。如果当前目录下没有logs目录，需要手动创建
|
|
logPath = ".\\logs\\"
|
|
# 日志文件名
|
|
print(logPath)
|
|
logName = os.path.join(logPath + loggerName + '.log')
|
|
print(logName)
|
|
|
|
# 创建一个每日可更新的handler，用于将日志输出文件
|
|
fileHandler = TimedRotatingFileHandler(logName, when='D',encoding='utf-8')
|
|
fileHandler.setLevel(logging.DEBUG)
|
|
|
|
# 创建一个handler，用于将日志输出到控制台
|
|
consoleHandler = logging.StreamHandler()
|
|
consoleHandler.setLevel(logging.DEBUG)
|
|
|
|
# 定义handler的输出格式
|
|
formatter = logging.Formatter(
|
|
'%(asctime)s-%(name)s-%(levelname)s:%(message)s')
|
|
fileHandler.setFormatter(formatter)
|
|
consoleHandler.setFormatter(formatter)
|
|
|
|
self.logger.addHandler(fileHandler)
|
|
#如果不需要控制台，可以注释掉
|
|
self.logger.addHandler(consoleHandler)
|
|
|
|
|
|
def getLogger(self):
|
|
return self.logger
|
|
log = cusLogger("rtsp_process").getLogger()
|
|
|
|
|
|
|
|
|
|
|
|
class rtspThread(threading.Thread):
|
|
def __init__(self, thName, rtspUrl, prefix, filePath, fps, frameSize, queueSize, eventKey, eventValue,DensityClass,DetectionClass,Fine_grianedClass):
|
|
threading.Thread.__init__(self)
|
|
# name
|
|
self.thName = thName
|
|
# rtsp流地址
|
|
self.rtspUrl = rtspUrl
|
|
# 帧率
|
|
self.fps = fps
|
|
# 帧大小（width,height)
|
|
self.frameSize = frameSize
|
|
# 存储队列大小，fps * 10秒
|
|
self.queueSize = queueSize
|
|
# redis的key
|
|
self.eventKey = eventKey
|
|
# key值
|
|
self.eventValue = eventValue
|
|
# 存储文件前缀
|
|
self.prefix = prefix
|
|
# 存储路径
|
|
self.filePath = filePath
|
|
self.DensityClass = DensityClass
|
|
self.DetectionClass = DetectionClass
|
|
self.Fine_grianedClass = Fine_grianedClass
|
|
|
|
def run(self):
|
|
vidoeSaveQueue = [queue.Queue(self.queueSize), queue.Queue(self.queueSize)]
|
|
jpgSaveQueue = [queue.Queue(self.queueSize), queue.Queue(self.queueSize)]
|
|
curQueueIdx = 0
|
|
t1 = time.time()
|
|
cap = cv2.VideoCapture(self.rtspUrl)
|
|
t2 = time.time()
|
|
print("capture is ",(t2-t1)*1000)
|
|
closed = False
|
|
failCnt = 0
|
|
winName = self.rtspUrl + "_" + str(time.time())
|
|
num_detection = 0
|
|
num_cuntupian=0
|
|
while True:
|
|
if cap.isOpened():
|
|
#t3 = time.time()
|
|
res, frame = cap.read()
|
|
#t4 = time.time()
|
|
#print("read is ",(t4-t3)*1000)
|
|
if res:
|
|
closed = False
|
|
failCnt = 0
|
|
# frame = cv2.imread("/home/wanghc/output1/0001_20210413155842/RawImage/0174.jpg")
|
|
# for i in range(100):
|
|
# t1 = time.time()
|
|
# print("initing.....")
|
|
# self.DetectionClass.check_object(frame)
|
|
# t2 = time.time()
|
|
# print((t2 - t1) * 1000)
|
|
if (num_detection >24) :
|
|
data_today = time.strftime('%H', time.localtime(time.time()))
|
|
# print(int(data_today))
|
|
if (int(data_today) > 7) & (int(data_today) < 18):
|
|
Check_start1 = Check_start_fordataselect(1,"check_start",frame,self.DetectionClass,self.Fine_grianedClass)
|
|
Check_start1.start()
|
|
else:
|
|
print("time is out")
|
|
num_detection=0
|
|
# if is_ex_bird :
|
|
# cv2.imwrite(self.filePath +"{}.jpg".format(time.strftime('%Y%m%d%H%M%S',time.localtime(time.time()))),frame)
|
|
num_detection+=1
|
|
num_cuntupian+=1
|
|
# cv2.namedWindow(winName, flags=cv2.WINDOW_FREERATIO)
|
|
# cv2.imshow(winName, frame)
|
|
# cv2.waitKey(10)
|
|
try:
|
|
if vidoeSaveQueue[curQueueIdx].full():
|
|
vidoeSaveQueue[curQueueIdx].get_nowait()
|
|
vidoeSaveQueue[curQueueIdx].put_nowait(frame)
|
|
except:
|
|
pass
|
|
try:
|
|
if jpgSaveQueue[curQueueIdx].full():
|
|
jpgSaveQueue[curQueueIdx].get_nowait()
|
|
jpgSaveQueue[curQueueIdx].put_nowait(frame)
|
|
except:
|
|
pass
|
|
|
|
# if (r_db.exists(self.eventKey) and int(r_db.get(self.eventKey)) == self.eventValue):
|
|
# print(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
|
|
# print(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
|
|
# r_db.delete(self.eventKey)
|
|
# # 存视频
|
|
# vidoeSaveProc = videoSaveThread(self.thName,self.prefix, self.filePath, self.fps,
|
|
# self.frameSize, vidoeSaveQueue[curQueueIdx])
|
|
# vidoeSaveProc.start()
|
|
# # 存图片
|
|
# # jpgSaveSaveProc = jpgSaveThread(self.thName, self.prefix, self.filePath, self.fps,
|
|
# # self.frameSize, jpgSaveQueue[curQueueIdx])
|
|
# # jpgSaveSaveProc.start()
|
|
# # 切换缓存队列
|
|
# curQueueIdx = (curQueueIdx + 1) % 2
|
|
|
|
else:
|
|
failCnt += 1
|
|
if(failCnt >= 15):
|
|
closed = True
|
|
else:
|
|
closed = True
|
|
if closed:
|
|
# 休眠10秒钟 重连
|
|
log.info("reconnect: " + self.rtspUrl)
|
|
time.sleep(5)
|
|
cap.open(self.rtspUrl)
|
|
|
|
if __name__ == '__main__':
|
|
|
|
|
|
|
|
|
|
root = "/home/wanghc/output_jiankong"
|
|
#r_db = redis.StrictRedis(host='localhost', port=6379, db=0)
|
|
log = cusLogger("rtsp_process").getLogger()
|
|
Density1 = Density(gp.devices, root)
|
|
Detection1 = Detection_Y5(gp.devices, root)
|
|
Fine_grained1 = Fine_grained(gp.devices, root)
|
|
# try:
|
|
# # 如果有多个流要同时取存，可以创建多个线程
|
|
# t1 = time.time()
|
|
# thread1 = rtspThread("thread_vs",
|
|
# "rtsp://admin:glodon123@192.168.116.13:554/Streaming/Channels/101",
|
|
# "ch1_",
|
|
# "../data_MPKFNet/RTSP/", # 为空串时，默认保存在当前目录
|
|
# 25,
|
|
# (3840, 2160),
|
|
# # (3264,2448),
|
|
# 1500,
|
|
# "key",
|
|
# 1,
|
|
# Density1,
|
|
# Detection1,
|
|
# Fine_grained1
|
|
# )
|
|
# t2 = time.time()
|
|
# print("multi is ", (t2 - t1) * 1000)
|
|
# thread1.start()
|
|
#
|
|
#
|
|
#
|
|
# except Exception as e:
|
|
# log.info(e)
|
|
try:
|
|
# 如果有多个流要同时取存，可以创建多个线程
|
|
t1 = time.time()
|
|
thread1 = rtspThread("thread_vs",
|
|
"rtsp://admin:glodon123@192.168.116.13:554/Streaming/Channels/101",
|
|
"ch1_",
|
|
"../data_MPKFNet/RTSP", # 为空串时，默认保存在当前目录
|
|
25,
|
|
(3840, 2160),
|
|
# (3264,2448),
|
|
1500,
|
|
"key",
|
|
1,
|
|
Density1,
|
|
Detection1,
|
|
Fine_grained1
|
|
)
|
|
t2 = time.time()
|
|
print("multi is ", (t2 - t1) * 1000)
|
|
thread1.start()
|
|
|
|
|
|
|
|
except Exception as e:
|
|
log.info(e)
|