57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
|
|
import os
|
|
import torch
|
|
from System_I import Density,Detection_Y5,Fine_grained,ThreadMonitor_mul,isopenmul
|
|
import cv2
|
|
import time
|
|
import redis
|
|
from rtsp_process import rtspThread,cusLogger
|
|
from common import global_para as gp
|
|
import globalvar as gl
|
|
|
|
r_db = None
|
|
#if __name__ == '__main__':
|
|
def Hatcher():
|
|
gl._init()
|
|
log = cusLogger("rtsp_process").getLogger()
|
|
|
|
Density1 = Density(gp.devices,gp.root)
|
|
Detection1 = Detection_Y5(gp.devices, gp.root)
|
|
Fine_grained1 = Fine_grained(gp.devices,gp.root)
|
|
# Density1.inference(image,image_time,str(1))
|
|
# Detection1.inference(image,image_time,str(1),Fine_grained1)
|
|
try:
|
|
# 如果有多个流要同时取存，可以创建多个线程
|
|
thread1 = rtspThread("thread_vs",
|
|
"rtsp://admin:glodon123@192.168.116.13:554/Streaming/Channels/101",
|
|
"ch1_",
|
|
"",# 为空串时，默认保存在当前目录
|
|
25,
|
|
(3840, 2160),
|
|
#(3264,2448),
|
|
150,
|
|
30,
|
|
"key",
|
|
1,
|
|
Density1,
|
|
Detection1,
|
|
Fine_grained1
|
|
)
|
|
|
|
thread1.start()
|
|
# ThreadMonitor_mul1 = ThreadMonitor_mul("0001","ThreadMonitor_mul") #开启线程管理
|
|
# ThreadMonitor_mul1.start()
|
|
|
|
|
|
|
|
|
|
|
|
except Exception as e:
|
|
log.info(e)
|
|
|
|
|
|
#
|
|
# Density1.inference(image,image_time,str(1))
|
|
# Detection1.inference(image,image_time,str(1),Fine_grained1)
|
|
#
|