基于生成对抗网络的异物检测方法
基本信息
申请号 | CN202111384661.5 | 申请日 | - |
公开(公告)号 | CN114220043A | 公开(公告)日 | 2022-03-22 |
申请公布号 | CN114220043A | 申请公布日 | 2022-03-22 |
分类号 | G06V20/40(2022.01)I;G06V10/82(2022.01)I;G06N3/04(2006.01)I;G06N3/08(2006.01)I | 分类 | 计算;推算;计数; |
发明人 | 刘伟铭;刘瑞康;汪良;毛良;邝雨婕 | 申请(专利权)人 | 广东珠三角城际轨道交通有限公司 |
代理机构 | 广州市华学知识产权代理有限公司 | 代理人 | 李斌 |
地址 | 510640广东省广州市天河区五山路381号 | ||
法律状态 | - |
摘要
摘要 | 本发明公开了一种基于生成对抗网络的异物检测方法,其特征在于,包括以下步骤:S1、将连续的多帧正常视频帧作为训练数据,按时间顺序堆叠起来输入到空洞细粒度生成器中重建当前帧进行训练;S2、构建图块级鉴别器,将被检测图像和重建图像分别输入到鉴别器中获得高维可区分的特征向量;S3、设计对抗性损失函数和局部注意重建损失函数;S4、根据损失函数对空洞细粒度生成器和图块级鉴别器进行对抗训练;S5、根据检测图片和重建图片在特征空间和图像空间上的加权重建误差构建异常分数特征图;设置一个异常阈值;S6、异常分数特征图中分数值大于阈值的区域标注为异物出现的位置。本发明适合于轨道交通站台门与列车之间的异物检测。 |
