wangjin0928 523723b694 udpate cityMangement2 jointUtils | 1 рік тому | |
---|---|---|
DMPRUtils | 1 рік тому | |
__pycache__ | 1 рік тому | |
conf | 1 рік тому | |
models | 1 рік тому | |
obbUtils | 1 рік тому | |
ocrUtils | 1 рік тому | |
ocrUtils2 | 1 рік тому | |
segutils | 1 рік тому | |
trackUtils | 1 рік тому | |
utils | 1 рік тому | |
utilsK | 1 рік тому | |
voduploadsdk | 1 рік тому | |
weights/conf | 1 рік тому | |
AI.py | 1 рік тому | |
DMPR.py | 1 рік тому | |
readme.md | 1 рік тому | |
stdc.py | 1 рік тому |
2.0--变化如下: 每个模型都要采用TRT模型 2023.1.17 1.增加检测模型(countryRoad)--乡间道路检测---类别包括:违法种植 2.增加钓鱼游泳模型(AnglerSwimmer)--类别包括:钓鱼、游泳 3.更新车辆检测模型--增加网络公开卡口数据 4.更新高速路里的公路检测分割模型--增加乡间小路的数据 2023.1.29 1.增加模型对4080的TRT兼容 2023.1.31 1.增加forest2业务,单独模型,类别包括:”林斑”,“病死树”,“人”,“火焰”,“烟雾” 2023.1.20 1.高速公路模型,增加车辆、行人检测。命名highWay2,只检测路上的目标,非路上的目标排除 2023.3.07 1.把所有模型都增A10显卡版本一份 2023.4.02 1.增加高速事故的检测
2023.4.11 1.增加OCR en模型 2.修改把事故检测V2添加到highWay2
2023.4.22-23 1.更新事故检测模型,修复BUG。事故不能单独检测,一定要和车辆一起检测。 2.增加倾斜船只检测模型,ship2。 3.开放分割结果的轮廓画框接口。用户决定是否画道路、水轮廓。line169:‘segLineShow’:True,来控制。
2023.5.5 1.增加“chanelEmerency”河道落水人员检测 2.增加OCR2模型,换了新的CRNN识别模型。 2023.6.9 1.更新张建川新的河道分割模型,放在业务“river2” 2023.7.3 1.添加城管项目,纯检测“车辆”、“垃圾”,业务名称”cityMangement”
2023.7.10 1.demo.py中添加drowing,落水人员检测。 2.demo.py中添加noParking 1.0版本,不成熟,后面还要更新。 3.demo3.0.py中对’river’, ‘highWay2’,‘noParking’,‘drowning’,‘forest2’,‘vehicle’,‘pedestrian’,‘smogfire’ , ‘AnglerSwimmer’,‘channelEmergency’, ‘countryRoad’,‘cityMangement’,增加跟踪功能。
2023.7.17 1.demo.py,demo3.0.py 的cityMangement模型增加“流动商贩”类别,修改后的模型检测类别是:“车辆”、“垃圾”、“流动商贩”
2023.7.28 1.demo.py 增加了illParking,城市违停车辆检测,检测模型输入三类(车、T角点,L角点),后处理变成“违停”.综合起来,在检测模型中有四类输出(车、T角点,L角点,违停 )
但是最终输出只有“违停”,类别好"3",其它车辆等不会输出。
2023.08.07 1.调整输出DSP的数据格式,demo.py和demo3.0.py都变。 矩形框(cls,x0,y0,x1,y1,score)-->(x0,y0,x1,y1,score,cls),倾斜框变为 [ (x0,y0),(x1,y1),(x2,y2),(x3,y3),score,cls] 2.完成切斜目标的跟踪。
2023.08.09 1.更新highWay2代码,修改输入参数,增加vehicleFlag,distanceFlag ,二者都为False时检测和分割混合的后处理最快。
实测2.png,检测和分割混合预处理时间如下:
vehicleFlag distanceFlag time(ms)
True True 2.55
True False 2.09
False True 1.56
False False 1.09
2.更新了Noparking(高速路车辆检测,0-常规车辆,1-应急车道上的车),规范了代码编写。 2023.08.14 1.修复highway2 vehicleFlag,distanceFlag 都为false时的bug 2023.08.20 1.增加cityMangement2 模型,包括了替代原来的的cityMangement和illParking,实现“车辆”、“垃圾”、“流动商贩”、“违停”,四个类别。 2.修改架构,segmodel由原来的demo中加载,改为在demo加载segmodel类,实现trt,pth统一加载,用segmodel.eval()实现统一预测。
该类别不仅适用于分类,还适用于dmpr模型,方便后续适用其它类别模型。
2023.08.22 1.增加cityRoad模型,单一检测模型,”护栏”,“交通标志”,“非交通标志”,“锥桶”,“水马”,其中”施工“第4,第5类别合并都叫”锥桶”,“水马”,名称相同. 2.对cityMangement2中yolo和dmpr融合处理debug。 2023.08.23 1.优化highWay2.0中yolo和stdc融合处理,使得分割模型输入的(640,360)-->(1920,1080)时候,时间增长不明显约为4.0ms
同时优化stdc的预处理部分改为torch处理,取代原来的numpy处理,现在两种尺寸的时间如下:
分割预处理 推理 分割检测融合处理
640*360 2.0 1.9 1.29
1920*1080 6.5 2.6 0.86
2023.08.29 1.更新drowning中的分割模型。 2.highWay2中添加了1920*1080的分割trt模型 2023.09.07 1.更新了cityRoad的检测模型 2023.09.15 1.demo3.0.py中增加了illparking的跟踪模型。这个模型不怎么用,只检测“违停”,和cityManget2里的功能重叠。 2.cityMangement2中违停dmpr模型更新,换成了backbone为yolov5s的模型,且检测yolov5模型权重也更新了。