19 changed files with 474 additions and 399 deletions
@ -0,0 +1,63 @@ |
|||
from myutils.ConfigManager import myCongif |
|||
if myCongif.get_data("model_platform") == "acl": |
|||
import acl |
|||
|
|||
SUCCESS = 0 # 成功状态值 |
|||
FAILED = 1 # 失败状态值 |
|||
|
|||
class ACLModeManger: |
|||
def __init__(self,): |
|||
self.acl_ok = False |
|||
|
|||
def __del__(self): |
|||
pass |
|||
|
|||
#初始化acl相关资源--一个进程内只能调用一次acl.init接口 |
|||
@staticmethod |
|||
def init_acl(device_id): |
|||
# '''acl初始化函数''' |
|||
ret = acl.init() # 0-成功，其它失败 |
|||
if ret: |
|||
raise RuntimeError(ret) |
|||
|
|||
ret = acl.rt.set_device(device_id) # 指定当前进程或线程中用于运算的Device。可以进程或线程中指定。*多设备时可以放线程* |
|||
# 在某一进程中指定Device，该进程内的多个线程可共用此Device显式创建Context（acl.rt.create_context接口）。 |
|||
if ret: |
|||
raise RuntimeError(ret) |
|||
print('ACL init Device Successfully') |
|||
|
|||
return True |
|||
|
|||
#去初始化 |
|||
@staticmethod |
|||
def del_acl(device_id): |
|||
'''Device去初始化''' |
|||
ret = acl.rt.reset_device(device_id) # 释放Device |
|||
if ret: |
|||
raise RuntimeError(ret) |
|||
|
|||
ret = acl.finalize() # 去初始化 0-成功，其它失败 --官方文档不建议放析构函数中执行 |
|||
if ret: |
|||
raise RuntimeError(ret) |
|||
print('ACL finalize Successfully') |
|||
return True |
|||
|
|||
@staticmethod |
|||
def th_inti_acl(device_id): |
|||
# 线程申请context |
|||
context, ret = acl.rt.create_context(device_id) # 显式创建一个Context |
|||
if ret: |
|||
raise RuntimeError(ret) |
|||
print('Init TH-Context Successfully') |
|||
return context |
|||
|
|||
@staticmethod |
|||
def th_del_acl(context): |
|||
#线程释放context |
|||
ret = acl.rt.destroy_context(context) # 释放 Context |
|||
if ret: |
|||
raise RuntimeError(ret) |
|||
print('Deinit TH-Context Successfully') |
|||
return True |
|||
|
|||
|
|||
@ -0,0 +1,120 @@ |
|||
import threading |
|||
from collections import deque |
|||
import numpy as np |
|||
import time |
|||
import copy |
|||
import queue |
|||
|
|||
class ChannelData: |
|||
def __init__(self, str_url, int_type, bool_run, deque_length,icount_max): |
|||
self.str_url = str_url #视频源地址 |
|||
self.int_type = int_type #视频源类型，0-usb,1-rtsp,2-hksdk |
|||
self.bool_run = bool_run #线程运行标识 |
|||
self.deque_frame = deque(maxlen=deque_length) |
|||
self.last_frame = None # 保存图片数据 |
|||
self.frame_queue = queue.Queue(maxsize=1) |
|||
self.counter = 0 #帧序列号--保存报警录像使用 |
|||
self.icount_max = icount_max #帧序列号上限 |
|||
self.lock = threading.RLock() # 用于保证线程安全 |
|||
|
|||
#添加一帧图片 |
|||
def add_deque(self, value): |
|||
self.deque_frame.append(value) #deque 满了以后会把前面的数据移除 |
|||
|
|||
#拷贝一份数据 |
|||
def copy_deque(self): |
|||
return copy.deepcopy(self.deque_frame) |
|||
|
|||
#获取最后一帧图片 |
|||
def get_last_frame(self): |
|||
with self.lock: |
|||
frame = self.last_frame |
|||
return frame |
|||
# if not self.frame_queue.empty(): |
|||
# return self.frame_queue.get() |
|||
# else: |
|||
# return None |
|||
|
|||
|
|||
def update_last_frame(self,buffer): |
|||
if buffer: |
|||
with self.lock: |
|||
self.last_frame = None |
|||
self.last_frame = buffer |
|||
# if not self.frame_queue.full(): |
|||
# self.frame_queue.put(buffer) |
|||
# else: |
|||
# self.frame_queue.get() # 丢弃最旧的帧 |
|||
# self.frame_queue.put(buffer) |
|||
|
|||
|
|||
#帧序列号自增 一个线程中处理，不用加锁 |
|||
def increment_counter(self): |
|||
self.counter += 1 |
|||
if self.counter > self.icount_max: |
|||
self.counter = 0 |
|||
|
|||
def get_counter(self): |
|||
return self.counter |
|||
|
|||
#清空数据，主要是删除deque 和 last_frame |
|||
def clear(self): |
|||
with self.lock: |
|||
self.bool_run = False |
|||
time.sleep(1) #休眠一秒，等待通道对应的子线程，停止工作。 |
|||
self.deque_frame.clear() |
|||
self.last_frame = None |
|||
|
|||
def stop_run(self): |
|||
self.bool_run = False |
|||
|
|||
|
|||
class ChannelManager: |
|||
def __init__(self): |
|||
self.channels = {} |
|||
self.lock = threading.RLock() # 用于保证字典操作的线程安全 |
|||
|
|||
#增加节点 |
|||
def add_channel(self, channel_id, str_url, int_type, bool_run, deque_length=10,icount_max=100000): |
|||
with self.lock: |
|||
if channel_id in self.channels: #若已经有数据，先删除后再增加 |
|||
self.channels[channel_id].clear() # 手动清理资源 |
|||
del self.channels[channel_id] |
|||
self.channels[channel_id] = ChannelData(str_url, int_type, bool_run, deque_length,icount_max) |
|||
|
|||
#删除节点 |
|||
def delete_channel(self, channel_id): |
|||
with self.lock: |
|||
if channel_id in self.channels: |
|||
self.channels[channel_id].clear() # 手动清理资源 |
|||
del self.channels[channel_id] |
|||
|
|||
#获取节点 |
|||
def get_channel(self, channel_id): |
|||
with self.lock: |
|||
return self.channels.get(channel_id) |
|||
|
|||
#停止工作线程 |
|||
def stop_channel(self,channel_id): |
|||
with self.lock: |
|||
if channel_id == 0: |
|||
for clannel_id,clannel_data in self.channels.items(): |
|||
clannel_data.clear() |
|||
del self.channels |
|||
else: |
|||
if channel_id in self.channels: |
|||
self.channels[channel_id].clear() # 手动清理资源 |
|||
del self.channels[channel_id] |
|||
|
|||
if __name__ == "__main__": |
|||
# 示例使用 |
|||
manager = ChannelManager() |
|||
manager.add_channel('channel_1', 'test', 123, True, deque_length=5) |
|||
|
|||
# 更新和读取操作 |
|||
manager.update_channel_deque('channel_1', 'data1') |
|||
manager.update_channel_buffer('channel_1', np.array([[1, 2], [3, 4]])) |
|||
manager.increment_channel_counter('channel_1') |
|||
|
|||
channel_data = manager.get_channel_data('channel_1') |
|||
print(channel_data) |
|||
@ -1,49 +0,0 @@ |
|||
<!DOCTYPE html> |
|||
<html lang="en"> |
|||
<head> |
|||
<meta charset="UTF-8"> |
|||
<meta name="viewport" content="width=device-width, initial-scale=1.0"> |
|||
<title>YOLOv5 Video Streams</title> |
|||
<style> |
|||
body { |
|||
display: flex; |
|||
justify-content: center; |
|||
align-items: center; |
|||
height: 100vh; |
|||
margin: 0; |
|||
background-color: #f0f0f0; |
|||
} |
|||
.grid-container { |
|||
display: grid; |
|||
grid-template-columns: repeat(3, 1fr); |
|||
grid-template-rows: repeat(3, 1fr); |
|||
gap: 10px; |
|||
width: 90vw; |
|||
height: 90vh; |
|||
} |
|||
.grid-item { |
|||
position: relative; |
|||
width: 100%; |
|||
padding-top: 75%; /* 4:3 aspect ratio */ |
|||
} |
|||
.grid-item img { |
|||
position: absolute; |
|||
top: 0; |
|||
left: 0; |
|||
width: 100%; |
|||
height: 100%; |
|||
object-fit: cover; |
|||
border-radius: 8px; |
|||
} |
|||
</style> |
|||
</head> |
|||
<body> |
|||
<div class="grid-container"> |
|||
{% for stream_id in streams %} |
|||
<div class="grid-item"> |
|||
<img src="{{ url_for('video_feed', stream_id=stream_id) }}" alt="Stream {{ stream_id }}"> |
|||
</div> |
|||
{% endfor %} |
|||
</div> |
|||
</body> |
|||
</html> |
|||
Binary file not shown.
Loading…
Reference in new issue