一种宠物识别方法

基本信息

申请号 CN202110664570.0 申请日 -
公开(公告)号 CN113392768A 公开(公告)日 2021-09-14
申请公布号 CN113392768A 申请公布日 2021-09-14
分类号 G06K9/00(2006.01)I;G06K9/46(2006.01)I;G06T7/90(2017.01)I;G06T7/194(2017.01)I 分类 计算;推算;计数;
发明人 蔡子轩;盛鹏;乔国坤 申请(专利权)人 新疆爱华盈通信息技术有限公司
代理机构 - 代理人 -
地址 830000新疆维吾尔自治区乌鲁木齐市经济技术开发区喀纳斯湖北路455号新疆软件园创智大厦A座10层1010-1011室
法律状态 -

摘要

摘要 本发明公开了一种宠物识别方法,使用yolov5作为宠物识别模型,使用语义分割模型的方式实现宠物素材的扩增,对扩增后的图片进行数据增强,对数据增强后的图片进行色度、亮度和对比度的调整,将调整后的图片输入宠物识别模型并使用Generalized Focal Loss损失函数来训练宠物识别模型,得到训练好的宠物识别模型,利用训练好的模型实现对宠物的识别。本发明能够实现对较大范围尺寸的宠物的准确识别,避免实际中宠物遮挡和姿态众多对识别的影响。