8 changed files with 179 additions and 83 deletions
@ -0,0 +1,76 @@ |
|||||
|
import multiprocessing |
||||
|
|
||||
|
from threading import Lock |
||||
|
from core.ModelNode import ModelNode |
||||
|
from myutils.ConfigManager import myCongif |
||||
|
|
||||
|
class ModelNodeManger: |
||||
|
def __init__(self,device_id,model_id,model_path,threshold=0.5,iou_thres=0.5): |
||||
|
self.ch_count = 0 #当前模型关联的通道数量 |
||||
|
self.channel_list = {} #channel_id modelNodle |
||||
|
self.clist_Lock = Lock() # count的维护锁 |
||||
|
self.modelNode_list = [] # model_Node |
||||
|
self.brun = multiprocessing.Value('b',True) #brun.value = False,brun.value = True |
||||
|
self.m_p_status = multiprocessing.Value('i',0) |
||||
|
self.model = None # 模型对象 |
||||
|
#------- |
||||
|
self.max_count = myCongif.get_data('maxCount') |
||||
|
self.device_id = device_id |
||||
|
self.model_id = model_id |
||||
|
self.model_path = model_path |
||||
|
self.threshold = threshold |
||||
|
self.iou_thres = iou_thres |
||||
|
|
||||
|
|
||||
|
# #添加数据 |
||||
|
# def pro_add_data(self,channel_id,data): |
||||
|
# pass |
||||
|
# # #加锁？ -- 对于多线程这里的耗时会不会有点大。。。 model_nodel 反馈出去？？？？？getModle_Nodel |
||||
|
# # with self.clist_Lock: |
||||
|
# # if channel_id in self.channel_list: |
||||
|
# # self.modelNode_list[channel_id].pro_add_data(data) |
||||
|
|
||||
|
#暴露modelNodel -- 避免pro_add_data锁竞争 |
||||
|
def getModle_Nodel(self,channel_id): |
||||
|
model_nodel = None |
||||
|
if channel_id in self.channel_list: |
||||
|
model_nodel = self.channel_list[channel_id] |
||||
|
return model_nodel |
||||
|
|
||||
|
def start_model_th(self, channel_id, out_mq): |
||||
|
with self.clist_Lock: |
||||
|
if channel_id in self.channel_list: #已经在List 说明已经启动过了，但不应该重复调用 |
||||
|
print("不应该重复调用start_model_th") |
||||
|
return True |
||||
|
else: # 还没入list ，查找可以插入的model_node |
||||
|
for model_node in self.modelNode_list: |
||||
|
if model_node.ch_count < self.max_count: # 没满可以插入 |
||||
|
model_node.start_model_th(channel_id, out_mq) #+1 |
||||
|
self.channel_list[channel_id] = model_node #新增 channel_di 与model_node的对应关系 |
||||
|
return True |
||||
|
# 执行到这说明没有空的model_nodel -- 需要新建 |
||||
|
model_node = ModelNode(self.device_id,self.model_path,self.max_count) |
||||
|
model_node.start_model_th(channel_id, out_mq) # 启动线程，并+1 |
||||
|
#两个数据结构，新增节点 |
||||
|
self.modelNode_list.append(model_node) #新增一个model_node节点 |
||||
|
self.channel_list[channel_id] = model_node #新增 channel_di 与model_node的对应关系 |
||||
|
#行动完更新数量 |
||||
|
self.ch_count +=1 |
||||
|
return True |
||||
|
|
||||
|
def stop_model_th(self, channel_id): |
||||
|
with self.clist_Lock: |
||||
|
if channel_id in self.channel_list: |
||||
|
model_node = self.channel_list[channel_id] |
||||
|
model_node.stop_model_th(channel_id) #减-，根据情况停止工作进程、 |
||||
|
# 需要考虑下model_node是否要删除，还是保留后重复利用- |
||||
|
if model_node.ch_count ==0: |
||||
|
self.modelNode_list.remove(model_node) #如何没有关联通道了，侧删除该modelnodel |
||||
|
del model_node #删除对象 |
||||
|
del self.channel_list[channel_id] #删除channel_id 与 model_node的对应关心节点 |
||||
|
#行动完更新数量 |
||||
|
self.ch_count -= 1 |
||||
|
return True |
||||
|
else: |
||||
|
print("不在list里面，不应该调用stop_model_th") |
||||
|
return False |
||||
Loading…
Reference in new issue