[发明专利]一种影像分类方法和系统有效
申请号: | 201610438796.8 | 申请日: | 2016-06-17 |
公开(公告)号: | CN107516103B | 公开(公告)日: | 2020-08-25 |
发明(设计)人: | 石建萍;李聪 | 申请(专利权)人: | 北京市商汤科技开发有限公司 |
主分类号: | G06K9/62 | 分类号: | G06K9/62;G06N20/00 |
代理公司: | 北京金阙华进专利事务所(普通合伙) 11224 | 代理人: | 陈建春 |
地址: | 100084 北京市海淀区中*** | 国省代码: | 北京;11 |
权利要求书: | 查看更多 | 说明书: | 查看更多 |
摘要: | |||
搜索关键词: | 一种 影像 分类 方法 系统 | ||
1.一种影像分类方法,其特征在于,所述方法包括:
将待分类的影像输入训练好的卷积神经网络,得到所述影像的每一像素对应的特征,所述每一像素对应的特征视为所述每一像素属于各个预设类别的预测概率;
将所述每一像素对应的特征及所述影像的原始影像通过条件随机场CRF模型处理对所述预测概率进行调整,得到所述影像的分类结果;
其中所述卷积神经网络包括依次对所述影像进行处理的多级处理级,每一处理级包括顺次连接的卷积层、归一化处理层和非线性变换处理层,每一在后的层的输入为上一层输出的特征图;其中部分所述处理级之间还包括池化层;其中所述多级处理级中的最后一级处理级的输出特征图再至少经卷积操作和插值操作得到所述影像的每一像素对应的特征。
2.根据权利要求1所述的方法,其特征在于,所述CRF模型通过计算像素点之间的颜色和空间位置的关系调整所述预测概率。
3.根据权利要求2所述的方法,其特征在于,所述CRF模型由下式表达:
其中i、j表示所述影像上的不同位置;xi包含像素i处的位置、颜色、特征信息;θi(xi)=-log P(xi)为一维势函数,P(xi)为像素点x在位置i处的类别概率,为二维势函数;当xi≠xj时,μ(xi,xj)=1,否则μ(xi,xj)=0;k为待分类类别数量,wm为高斯核函数km(fi,fj)的权重系数,fi表示像素点x在位置i处的特征信息。
4.根据权利要求1所述的方法,其特征在于,所述多级处理级包括7-14级处理级。
5.根据权利要求1所述的方法,其特征在于,所述多级处理级中除最后一级处理级之外的每一处理级的卷积层的卷积核大小为3×3,步长为2。
6.根据权利要求1所述的方法,其特征在于,所述池化层的卷积核大小为3×3。
7.一种影像分类系统,其特征在于,所述系统包括:
影像输入及概率确定模块,用于将待分类的影像输入训练好的卷积神经网络,得到所述影像的每一像素对应的特征,所述每一像素对应的特征视为所述每一像素属于各个预设类别的预测概率;
分类确定模块,用于将所述每一像素对应的特征及所述影像的原始影像通过条件随机场CRF模型处理对所述预测概率进行调整,得到所述影像的分类结果;
其中所述卷积神经网络包括依次对所述影像进行处理的多级处理级,每一处理级包括顺次连接的卷积层、归一化处理层和非线性变换处理层,每一在后的层的输入为上一层输出的特征图;其中部分所述处理级之间还包括池化层;其中所述多级处理级中的最后一级处理级的输出特征图再至少经卷积操作和插值操作得到所述影像的每一像素对应的特征。
8.根据权利要求7所述的系统,其特征在于,所述CRF模型通过计算像素点之间的颜色和空间位置的关系调整所述预测概率。
该专利技术资料仅供研究查看技术是否侵权等信息,商用须获得专利权人授权。该专利全部权利属于北京市商汤科技开发有限公司,未经北京市商汤科技开发有限公司许可,擅自商用是侵权行为。如果您想购买此专利、获得商业授权和技术合作,请联系【客服】
本文链接:http://www.vipzhuanli.com/pat/books/201610438796.8/1.html,转载请声明来源钻瓜专利网。