1)新增M027:建筑物下行人检测及计数 2)密集人群计数模型及自研车牌模型优化 3)分类模型支持pt模型加载(巴中水利)
This commit is contained in:
parent
b75a74d52c
commit
585a7a05f7
|
|
@ -406,13 +406,8 @@ class DENSECROWDCOUNTModel:
|
|||
par = modeType.value[4](str(device), gpu_name)
|
||||
rainbows = par["rainbows"]
|
||||
models=[ modelPar['model'](weights=modelPar['weight'],par=modelPar['par']) for modelPar in par['models'] ]
|
||||
<<<<<<< HEAD
|
||||
postPar = [pp['par'] for pp in par['models']]
|
||||
self.model_conf = (modeType, device, models, postPar, rainbows)
|
||||
=======
|
||||
postPar = par['models'][0]['par']
|
||||
self.model_conf = (modeType, device, models[0], postPar, rainbows)
|
||||
>>>>>>> origin/zsl
|
||||
except Exception:
|
||||
logger.error("模型加载异常:{}, requestId:{}", format_exc(), requestId)
|
||||
raise ServiceException(ExceptionType.MODEL_LOADING_EXCEPTION.value[0],
|
||||
|
|
@ -761,7 +756,6 @@ MODEL_CONFIG = {
|
|||
None,
|
||||
lambda x: cc_process(x)
|
||||
),
|
||||
<<<<<<< HEAD
|
||||
# 加载建筑物下行人检测模型
|
||||
ModelType.CITY_UNDERBUILDCOUNT_MODEL.value[1]: (
|
||||
lambda x, y, r, t, z, h: DENSECROWDCOUNTModel(x, y, r, ModelType.CITY_UNDERBUILDCOUNT_MODEL, t, z, h),
|
||||
|
|
@ -769,6 +763,4 @@ MODEL_CONFIG = {
|
|||
None,
|
||||
lambda x: cc_process(x)
|
||||
),
|
||||
=======
|
||||
>>>>>>> origin/zsl
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue